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))