Spaces:
Running
Running
File size: 1,328 Bytes
2797a7a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
import unittest
from pathlib import Path
from aistorybooks.phidata.classic_stories import PhiStoryBookGenerator
@unittest.skip("Local user test only")
class TestPhiStoryBookGenerator(unittest.TestCase):
def test_story_book_generator(self):
pdf_file = Path(__file__).parent.joinpath("resources/LoremIpsum.pdf")
generator = PhiStoryBookGenerator(
language="German",
level="A1 Intermediate",
summary_size="Long (150 sentences/1200 words)",
writing_style="Philosophical",
)
results = generator.run(pdf_file=pdf_file, chunk_size=1, padding=0, skip_first_n_pages=0)
i = 0
for result in results:
i += 1
print(result.metrics)
if i > 2:
raise "STOP"
@unittest.skip("Local user test only")
def test_story_book_generator_run_chunk(self):
generator = PhiStoryBookGenerator(
language="German",
level="A1 Intermediate",
summary_size="Long (150 sentences/1200 words)",
writing_style="Funny",
)
result = generator.author_agent.run("hello, how are you doing")
print(result)
result = generator._run_chunk(content="hello, how are you doing", start_page=0, end_page=1)
print(result)
|