fix: infer filing status from us-cpa questions

This commit is contained in:
Stefano Fiorini
2026-03-15 04:20:43 -05:00
parent 12838f7449
commit f219672a2e
2 changed files with 51 additions and 1 deletions

View File

@@ -37,6 +37,34 @@ class QuestionEngineTests(unittest.TestCase):
self.assertTrue(analysis["authorities"])
self.assertEqual(analysis["authorities"][0]["sourceClass"], "irs_instructions")
def test_standard_deduction_infers_married_filing_jointly_from_question(self) -> None:
with tempfile.TemporaryDirectory() as temp_dir:
engine = self.build_engine(temp_dir)
analysis = engine.answer(
question="What is the standard deduction for married filing jointly?",
tax_year=2025,
case_facts={},
)
self.assertEqual(analysis["issue"], "standard_deduction")
self.assertEqual(analysis["conclusion"]["answer"], "$31,500")
self.assertIn("Married Filing Jointly", analysis["conclusion"]["summary"])
def test_standard_deduction_infers_head_of_household_from_question(self) -> None:
with tempfile.TemporaryDirectory() as temp_dir:
engine = self.build_engine(temp_dir)
analysis = engine.answer(
question="What is the standard deduction for a head of household filer?",
tax_year=2025,
case_facts={},
)
self.assertEqual(analysis["issue"], "standard_deduction")
self.assertEqual(analysis["conclusion"]["answer"], "$23,625")
self.assertIn("Head Of Household", analysis["conclusion"]["summary"])
def test_complex_question_flags_primary_law_escalation(self) -> None:
with tempfile.TemporaryDirectory() as temp_dir:
engine = self.build_engine(temp_dir)