diff --git a/democrasite/webiscite/tests/test_templates.py b/democrasite/webiscite/tests/test_templates.py index 3b83910..bc27402 100644 --- a/democrasite/webiscite/tests/test_templates.py +++ b/democrasite/webiscite/tests/test_templates.py @@ -130,6 +130,7 @@ def test_my_bills(self, status: Bill.Status, view: str, bill: Bill, client: Clie assert bill.name in content assert "Log in to vote" not in content assert "vote.js" in content - assert (bill.get_status_display() in content) == ( - bill.status != Bill.Status.OPEN - ) + if bill.status == Bill.Status.OPEN: + assert bill.get_status_display() + "

" not in content + else: + assert bill.get_status_display() in content