def testGetTeamNames(self): xml = """ <league> <team id="t1"><name>Team 1</name></team> <team id="t2"><name>Team 2</name></team> <team id="t3"><name>Team 3</name></team> </league> """ leagueElement = ElementTree.fromstring(xml) generator = LeagueFixturesReportGenerator() result = generator.getTeamNames(leagueElement) expectedResult = {} for i in range(1, 4): expectedResult["t{0}".format(i)] = "Team {0}".format(i) self.assertEqual(len(expectedResult), len(result)) for k, v in expectedResult.items(): self.assertEqual(v, result[k])