You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
37 lines
907 B
37 lines
907 B
5 months ago
|
from ai_acceleerated_learning.Vocal import Vocal
|
||
|
|
|
||
|
|
vocal = Vocal()
|
||
|
|
|
||
|
|
|
||
|
|
def test_pass():
|
||
|
|
assert (
|
||
|
|
vocal.generate_video(
|
||
|
|
"I love to play basketball, and I am a very good player.",
|
||
|
|
"basketball",
|
||
|
|
)
|
||
|
|
== "Successfully generated a YouTube video for your prompt: I"
|
||
|
|
" love to play basketball, and I am a very good player."
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_invalid_sports():
|
||
|
|
assert (
|
||
|
|
vocal.generate_video("I just ate some delicious tacos", "tacos")
|
||
|
|
== "Invalid sports entered!! Please enter a valid sport."
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_invalid_prompt():
|
||
|
|
assert (
|
||
|
|
vocal.generate_video(987, "basketball")
|
||
|
|
== "Invalid prompt entered!! Please enter a valid prompt."
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_not_string():
|
||
|
|
assert (
|
||
|
|
vocal.generate_video(789, 234)
|
||
|
|
== "Invalid prompt and sports entered!! Please enter valid"
|
||
|
|
" prompt and sport."
|
||
|
|
)
|