def test_init_with_ids(self): ''' I'm not sure how to test classes like this. So, what tests are doing is basically: call method with some arguments, check if something goes wrong. ''' # this is fine Heroes.from_ids([1, 14, 112]) # call with invalid id self.assertRaises(ValueError, Heroes.from_ids, [0, 228]) # call with non-integer id self.assertRaises(TypeError, Heroes.from_ids, ['1', 54]) # call with specific patch, fine too Heroes.from_ids([15, 25, 89], patch='706f')
def test_iter(self): ''' One more time: I'm not sure what to do there, but user should be able to iterate over heroes. ''' heroes = Heroes.from_ids([1, 15, 65]) for hero in heroes: pass