def test_get_data(self): visitor1 = MaxTestCasesCount() visitor2 = MaxTestCasesCount() visitor1.visit(self.submit1) visitor1.visit(self.submit2) visitor2.visit(self.submit3) res1 = visitor1.get_stat_data() res2 = visitor2.get_stat_data() self.assertEqual(res1, 10) self.assertEqual(res2, 11)
def test_pretty_print(self): visitor = MaxTestCasesCount() visitor.result = 42 self.assertEqual(visitor.pretty_print(), 'Test cases:42')