| import pytest |
| from pydantic import ValidationError |
| from ankigen.models import ( |
| Step, |
| Subtopics, |
| Topics, |
| CardFront, |
| CardBack, |
| Card, |
| CardList, |
| ConceptBreakdown, |
| CardGeneration, |
| LearningSequence, |
| ) |
|
|
|
|
| def test_step_model(): |
| step = Step(explanation="expl", output="out") |
| assert step.explanation == "expl" |
| assert step.output == "out" |
|
|
| with pytest.raises(ValidationError): |
| Step(explanation="expl") |
|
|
|
|
| def test_subtopics_model(): |
| sub = Subtopics(steps=[Step(explanation="e", output="o")], result=["r1"]) |
| assert len(sub.steps) == 1 |
| assert sub.result == ["r1"] |
|
|
|
|
| def test_topics_model(): |
| topics = Topics(result=[Subtopics(steps=[], result=[])]) |
| assert len(topics.result) == 1 |
|
|
|
|
| def test_card_front_model(): |
| cf = CardFront(question="Q?") |
| assert cf.question == "Q?" |
| assert CardFront().question is None |
|
|
|
|
| def test_card_back_model(): |
| cb = CardBack(answer="A", explanation="E", example="Ex") |
| assert cb.answer == "A" |
| assert cb.explanation == "E" |
| assert cb.example == "Ex" |
|
|
|
|
| def test_card_model(): |
| cf = CardFront(question="Q") |
| cb = CardBack(answer="A", explanation="E", example="Ex") |
| card = Card(front=cf, back=cb, metadata={"key": "val"}, card_type="cloze") |
| assert card.front.question == "Q" |
| assert card.card_type == "cloze" |
| assert card.metadata["key"] == "val" |
|
|
|
|
| def test_card_list_model(): |
| cf = CardFront(question="Q") |
| cb = CardBack(answer="A", explanation="E", example="Ex") |
| card = Card(front=cf, back=cb) |
| cl = CardList(topic="test", cards=[card]) |
| assert cl.topic == "test" |
| assert len(cl.cards) == 1 |
|
|
|
|
| def test_concept_breakdown_model(): |
| cb = ConceptBreakdown( |
| main_concept="C", |
| prerequisites=["P"], |
| learning_outcomes=["L"], |
| difficulty_level="beginner", |
| ) |
| assert cb.difficulty_level == "beginner" |
|
|
|
|
| def test_card_generation_model(): |
| cf = CardFront(question="Q") |
| cb = CardBack(answer="A", explanation="E", example="Ex") |
| card = Card(front=cf, back=cb) |
| cg = CardGeneration( |
| concept="C", thought_process="T", verification_steps=["V"], card=card |
| ) |
| assert cg.concept == "C" |
| assert cg.card.front.question == "Q" |
|
|
|
|
| def test_learning_sequence_model(): |
| ls = LearningSequence( |
| topic="T", |
| concepts=[], |
| cards=[], |
| suggested_study_order=[], |
| review_recommendations=[], |
| ) |
| assert ls.topic == "T" |
|
|
