def test_write_read_raster_h5(self):
"""Write and read hdf5 format"""
file_name = str(DATA_DIR.joinpath('test_centr.h5'))
centr = Centroids()
xf_lat, xo_lon, d_lat, d_lon, n_lat, n_lon = 10, 5, -0.5, 0.2, 20, 25
centr.set_raster_from_pix_bounds(xf_lat, xo_lon, d_lat, d_lon, n_lat,
n_lon)
centr.write_hdf5(file_name)
centr_read = Centroids()
centr_read.read_hdf5(file_name)
self.assertTrue(centr_read.meta)
self.assertFalse(centr_read.lat.size)
self.assertFalse(centr_read.lon.size)
self.assertEqual(centr_read.meta['width'], centr.meta['width'])
self.assertEqual(centr_read.meta['height'], centr.meta['height'])
self.assertAlmostEqual(centr_read.meta['transform'].a,
centr.meta['transform'].a)
self.assertAlmostEqual(centr_read.meta['transform'].b,
centr.meta['transform'].b)
self.assertAlmostEqual(centr_read.meta['transform'].c,
centr.meta['transform'].c)
self.assertAlmostEqual(centr_read.meta['transform'].d,
centr.meta['transform'].d)
self.assertAlmostEqual(centr_read.meta['transform'].e,
centr.meta['transform'].e)
self.assertAlmostEqual(centr_read.meta['transform'].f,
centr.meta['transform'].f)
self.assertTrue(
u_coord.equal_crs(centr_read.meta['crs'], centr.meta['crs']))
def test_set_raster_pxl_pass(self):
""" Test set_raster_from_pxl_bounds from pixel borders """
centr = Centroids()
xf_lat, xo_lon, d_lat, d_lon, n_lat, n_lon = 10, 5, -0.5, 0.2, 20, 25
centr.set_raster_from_pix_bounds(xf_lat, xo_lon, d_lat, d_lon, n_lat, n_lon)
self.assertEqual(centr.meta['crs'], DEF_CRS)
self.assertEqual(centr.meta['width'], n_lon)
self.assertEqual(centr.meta['height'], n_lat)
self.assertAlmostEqual(centr.meta['transform'][0], d_lon)
self.assertAlmostEqual(centr.meta['transform'][1], 0.0)
self.assertAlmostEqual(centr.meta['transform'][2], xo_lon)
self.assertAlmostEqual(centr.meta['transform'][3], 0.0)
self.assertAlmostEqual(centr.meta['transform'][4], d_lat)
self.assertAlmostEqual(centr.meta['transform'][5], xf_lat)
self.assertTrue('lat' in centr.__dict__.keys())
self.assertTrue('lon' in centr.__dict__.keys())
