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)
