def test_get_next_coordinate_last_coordinate_includes_end(self): map_generator = UniformMapGenerator() coordinate_gen = map_generator.get_next_coordinate(0.9, 1, 0.2) self.assertEqual(coordinate_gen.next(), 0.9) self.assertEqual(coordinate_gen.next(), 1.1) self.assertRaises(StopIteration, coordinate_gen.next)
def test_get_next_coordinate_last_coordinate_doesnt_overshoot(self): map_generator = UniformMapGenerator() coordinate_gen = map_generator.get_next_coordinate(0.8, 1, 0.2) self.assertEqual(coordinate_gen.next(), 0.8) self.assertEqual(coordinate_gen.next(), 1) self.assertRaises(StopIteration, coordinate_gen.next)
def test_get_next_coordinate_first_two(self): map_generator = UniformMapGenerator() coordinate_gen = map_generator.get_next_coordinate(0.5, 5, 0.2) self.assertEqual(coordinate_gen.next(), 0.5) self.assertEqual(coordinate_gen.next(), 0.7)