class Test(unittest.TestCase): def setUp(self): self.data = DotaBuffData() self.match = Match(32, datetime(2014, 5, 12, 21, 30), (30, 0)) def testAddMatch(self): self.data.add_match(self.match) retrieved_match = self.data.get_match(self.match.match_id) self.assertEqual(self.match, retrieved_match) def testGetInexistentMatch(self): retrieved_match = self.data.get_match(self.match.match_id) self.assertEqual(None, retrieved_match)
def setUp(self): self.data = DotaBuffData() self.match = Match(32, datetime(2014, 5, 12, 21, 30), (30, 0))