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)