def test_union(self):
cent1 = Centroids()
cent1.lat, cent1.lon = np.array([0, 1]), np.array([0, -1])
cent1.on_land = np.array([True, True])
cent2 = Centroids()
cent2.lat, cent2.lon = np.array([2, 3]), np.array([-2, 3])
cent2.on_land = np.array([False, False])
cent3 = Centroids()
cent3.lat, cent3.lon = np.array([-1, -2]), np.array([1, 2])
cent = cent1.union(cent2)
np.testing.assert_array_equal(cent.lat, [0, 1, 2, 3])
np.testing.assert_array_equal(cent.lon, [0, -1, -2, 3])
np.testing.assert_array_equal(cent.on_land, [True, True, False, False])
cent = cent1.union(cent1, cent2)
np.testing.assert_array_equal(cent.lat, [0, 1, 2, 3])
np.testing.assert_array_equal(cent.lon, [0, -1, -2, 3])
np.testing.assert_array_equal(cent.on_land, [True, True, False, False])
cent = Centroids().union(cent1)
np.testing.assert_array_equal(cent.lat, [0, 1])
np.testing.assert_array_equal(cent.lon, [0, -1])
np.testing.assert_array_equal(cent.on_land, [True, True])
cent = cent1.union(cent1)
np.testing.assert_array_equal(cent.lat, [0, 1])
np.testing.assert_array_equal(cent.lon, [0, -1])
np.testing.assert_array_equal(cent.on_land, [True, True])
cent = Centroids().union(cent1, cent2, cent3)
np.testing.assert_array_equal(cent.lat, [0, 1, 2, 3, -1, -2])
np.testing.assert_array_equal(cent.lon, [0, -1, -2, 3, 1, 2])
def test_centroids_check_pass(self):
"""Test vector data in Centroids"""
data_vec = TestVector.data_vector()
centr = Centroids()
centr.lat, centr.lon, centr.geometry = data_vec
centr.check()
self.assertEqual(centr.crs, data_vec[2].crs)
self.assertIsInstance(centr.total_bounds, tuple)
for i in range(4):
self.assertEqual(centr.total_bounds[i],
data_vec[2].total_bounds[i])
self.assertIsInstance(centr.lat, np.ndarray)
self.assertIsInstance(centr.lon, np.ndarray)
self.assertIsInstance(centr.coord, np.ndarray)
self.assertTrue(np.array_equal(centr.lat, VEC_LAT))
self.assertTrue(np.array_equal(centr.lon, VEC_LON))
self.assertTrue(
np.array_equal(centr.coord,
np.array([VEC_LAT, VEC_LON]).transpose()))
self.assertEqual(centr.size, VEC_LON.size)
centr.area_pixel = np.array([1])
with self.assertRaises(ValueError):
centr.check()
centr.area_pixel = np.array([])
centr.dist_coast = np.array([1])
with self.assertRaises(ValueError):
centr.check()
centr.dist_coast = np.array([])
centr.on_land = np.array([1])
with self.assertRaises(ValueError):
centr.check()
centr.on_land = np.array([])
centr.region_id = np.array([1])
with self.assertRaises(ValueError):
centr.check()
centr.region_id = np.array([])
centr.lat = np.array([1])
with self.assertRaises(ValueError):
centr.check()
centr.lat = np.array([])
centr.lon = np.array([1])
with self.assertRaises(ValueError):
centr.check()
centr.lon = np.array([])
centr.geometry = gpd.GeoSeries(np.ones(1))
with self.assertRaises(ValueError):
centr.check()
def test_centroids_check_pass(self):
"""Test vector data in Centroids"""
data_vec = data_vector()
centr = Centroids()
centr.lat, centr.lon, centr.geometry = data_vec
centr.check()
self.assertTrue(u_coord.equal_crs(centr.crs, data_vec[2].crs))
self.assertIsInstance(centr.total_bounds, tuple)
for i in range(4):
self.assertEqual(centr.total_bounds[i],
data_vec[2].total_bounds[i])
self.assertIsInstance(centr.lat, np.ndarray)
self.assertIsInstance(centr.lon, np.ndarray)
self.assertIsInstance(centr.coord, np.ndarray)
self.assertTrue(np.array_equal(centr.lat, VEC_LAT))
self.assertTrue(np.array_equal(centr.lon, VEC_LON))
self.assertTrue(
np.array_equal(centr.coord,
np.array([VEC_LAT, VEC_LON]).transpose()))
self.assertEqual(centr.size, VEC_LON.size)
centr.area_pixel = np.array([1])
with self.assertRaises(ValueError):
centr.check()
centr.area_pixel = np.array([])
centr.dist_coast = np.array([1])
with self.assertRaises(ValueError):
centr.check()
centr.dist_coast = np.array([])
centr.on_land = np.array([1])
with self.assertRaises(ValueError):
centr.check()
centr.on_land = np.array([])
centr.region_id = np.array([1])
with self.assertRaises(ValueError):
centr.check()
centr.region_id = np.array([])
centr.lat = np.array([1])
with self.assertRaises(ValueError):
centr.check()
centr.lat = np.array([])
centr.lon = np.array([1])
with self.assertRaises(ValueError):
centr.check()
centr.lon = np.array([])
centr.geometry = gpd.GeoSeries(np.ones(1))
with self.assertRaises(ValueError):
centr.check()
cen = Centroids()
cen.meta = {
'width': 10,
'height': 20,
'transform': rasterio.Affine(0.1, 0, 0, 0, 0.1, 0),
'crs': DEF_CRS,
}
with self.assertRaises(ValueError):
cen.check()
