def test_get_number_of_intervals_returns_int(self):
generator = UniformMapGenerator()
self.assertIsInstance(generator.get_number_of_intervals(1, 1, 1),
int)
def test_get_uniform_coordinate_map_output_is_expected_length(self):
generator = UniformMapGenerator()
dataframe = generator.get_uniform_coordinate_map(1, 10,
6, 20,
1, 1)
self.assertEqual(len(dataframe), 150, "Output length should be 150")
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_uniform_coordinate_map_returns_pandas_dataframe(self):
generator = UniformMapGenerator()
dataframe = generator.get_uniform_coordinate_map(1, 1, 1, 1, 1, 1)
self.assertIsInstance(dataframe, pandas.DataFrame)
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)
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_empty_out_dataframe_has_correct_column_names(self):
generator = UniformMapGenerator()
dataframe = generator.instantiate_output_dataframe(20)
self.assertEqual(dataframe.columns[0], 'addr_lat')
self.assertEqual(dataframe.columns[1], 'addr_lon')
def test_empty_output_dataframe_is_correct_length(self):
generator = UniformMapGenerator()
dataframe = generator.instantiate_output_dataframe(20)
self.assertEqual(len(dataframe), 20)
def test_empty_output_dataframe_is_pandas_dataframe(self):
generator = UniformMapGenerator()
self.assertIsInstance(generator.instantiate_output_dataframe(1),
pandas.DataFrame)
def test_get_number_of_intervals_decimal_properties(self):
generator = UniformMapGenerator()
number_of_intervals = generator.get_number_of_intervals(2, 3, 0.3)
self.assertEqual(number_of_intervals, 5)
def test_get_number_of_intervals_integers(self):
generator = UniformMapGenerator()
number_of_intervals = generator.get_number_of_intervals(3, 10, 1)
self.assertEqual(number_of_intervals, 8)