def test_fill_missing_no_input(self): '''Test Configuration class.''' # Arrange config = hm.Configuration(use_defaults=True) # Act / Assert with self.assertRaisesRegex(ValueError, "no input specified"): config.fill_missing()
def test_basic(self): '''Test Configuration class.''' # Act config = hm.Configuration(use_defaults=True) # Assert self.assertEqual(config.margin, 0) self.assertEqual(config.frequency, 1)
def setup_config(count): config = hm.Configuration() config.shapes = shapes_generator(count) config.projection = hm.EquirectangularProjection() config.projection.pixels_per_degree = 30 config.decay = 1 config.kernel = hm.LinearKernel(5) config.background = 'black' config.fill_missing() return config
def test_choose_osm_zoom_with_zoom_in_config(self): # Arrange config = hm.Configuration(use_defaults=True) config.zoom = 2 padding = 2 # Act zoom = hm.choose_osm_zoom(config, padding) # Assert self.assertEqual(zoom, 2)
def test_choose_osm_zoom_with_no_zoom_w_h_in_config(self): # Arrange config = hm.Configuration(use_defaults=True) padding = 2 # Act # Act / Assert with self.assertRaisesRegex( ValueError, "For OSM, you must specify height, " "width, or zoom"): hm.choose_osm_zoom(config, padding)
def test_choose_osm_zoom_with_w_h_in_config(self): # Arrange config = hm.Configuration(use_defaults=True) config.width = 400 config.height = 200 config.extent_in = hm.Extent(coords=(hm.LatLon(-10, -10), hm.LatLon(10, 10))) padding = 2 # Act zoom = hm.choose_osm_zoom(config, padding) # Assert self.assertEqual(zoom, 3)
def setup_config(self, iter): map = PIL.Image.open(config["map_img"]) self.map_size = map.size self.hm_config = hm.Configuration() self.hm_config.width = map.size[0] self.hm_config.height = map.size[1] self.hm_config.margin = 0 self.hm_config.zoom = 1 self.hm_config.projection = hm.EquirectangularProjection() self.hm_config.projection.pixels_per_degree = 1 self.hm_config.extent_out = hm.Extent( coords=[hm.Coordinate(c[0], c[1]) for c in [(0, 0), map.size]]) self.hm_config.extent_in = hm.Extent(coords=[ self.hm_config.projection.inverse_project(hm.Coordinate( c[0], c[1])) for c in [(0, 0), map.size] ]) self.hm_config.shapes = iter self.hm_config.decay = 0.5 self.hm_config.kernel = hm.LinearKernel(5) self.hm_config.background_image = map self.hm_config.fill_missing() return config