o Lojøã@s>ddlZddlmZGdd„dejƒZedkre ¡dSdS)éN)Ú build_promptc@s$eZdZdd„Zdd„Zdd„ZdS)Ú GeneratorTestcCs<tdddddœgƒ}| d|¡| d|¡| d|¡dS) NzWhat is overtime?ÚFLSAz+Overtime is paid at one and one-half times.úhttps://example.test/flsa©ÚsourceÚtextÚurlz[1]zCite every factual claimzNOT legal advice©rÚassertIn©ÚselfÚprompt©rú2/root/openrights-assistant/tests/test_generator.pyÚ)test_prompt_contains_citation_instructions  z7GeneratorTest.test_prompt_contains_citation_instructioncCs$tdddddœgƒ}| d|¡dS)NzWhat is the minimum wage?rzThe minimum wage is $7.25.rrr r rrrÚtest_prompt_includes_question óz+GeneratorTest.test_prompt_includes_questioncCs$tdddddœgƒ}| d|¡dS)Nztest?ÚTestzUnique passage text here.zhttps://example.testrr r rrrÚ test_prompt_includes_source_textrz.GeneratorTest.test_prompt_includes_source_textN)Ú__name__Ú __module__Ú __qualname__rrrrrrrrs rÚ__main__)ÚunittestÚopenrights.generatorrÚTestCaserrÚmainrrrrÚs   ÿ