def test_get_border_dateline(self): dom = Domain('+proj=stere +datum=WGS84 +ellps=WGS84 +lat_0=0 +lon_0=-179 +no_defs', '-te -1000000 -1000 1000000 1000 -ts 10 10') brd = dom.get_border() self.assertTrue(np.all(brd[0] > 0)) brd = dom.get_border(fix_lon=False) self.assertTrue(brd[0].min() < 0)
def test_write_domain_map__basemap_missing(self): plt.switch_backend('Agg') d = Domain(4326, "-te 25 70 35 72 -ts 500 500") border = d.get_border() tmpfilename = os.path.join(ntd.tmp_data_path, 'domain_write_map.png') with self.assertRaises(ImportError): write_domain_map(border, tmpfilename, labels=['Patch1'])
def test_get_border(self): d = Domain(4326, "-te 25 70 35 72 -ts 500 500") lon, lat = d.get_border() self.assertEqual(type(lon), np.ndarray) self.assertEqual(type(lat), np.ndarray) self.assertEqual(len(lat), 44) self.assertEqual(lat[0], lat[-1]) self.assertEqual(lon[0], lon[-1])
def test_write_domain_map(self): plt.switch_backend('Agg') d = Domain(4326, "-te 25 70 35 72 -ts 500 500") border = d.get_border() tmpfilename = os.path.join(ntd.tmp_data_path, 'domain_write_map.png') write_domain_map(border, tmpfilename, labels=['Patch1']) self.assertTrue(os.path.exists(tmpfilename)) i = Image.open(tmpfilename) i.verify() self.assertEqual(i.info['dpi'], (50, 50))
def test_write_domain_map_labels(self): plt.switch_backend('Agg') d = Domain(4326, "-te 25 70 35 72 -ts 500 500") border = d.get_border() tmpfilename = os.path.join(ntd.tmp_data_path, 'domain_write_map_labels.png') write_domain_map(border, tmpfilename, mer_labels=[False, False, False, True], par_labels=[True, False, False, False]) self.assertTrue(os.path.exists(tmpfilename)) i = Image.open(tmpfilename) i.verify()
def test_get_border(self): dom = Domain(4326, "-te 25 70 35 72 -ts 500 500") result = dom.get_border(nPoints=10) lat, lon = result self.assertEqual(type(lat), np.ndarray) self.assertEqual(type(lon), np.ndarray) self.assertIsInstance(result, tuple) self.assertEqual(len(result), 2) test_x = [25., 26., 27., 28., 29., 30., 31., 32., 33., 34., 35., 35., 35., 35., 35., 35., 35., 35., 35., 35., 35., 35., 35., 34., 33., 32., 31., 30., 29., 28., 27., 26., 25., 25., 25., 25., 25., 25., 25., 25., 25., 25., 25., 25.] test_y = [72., 72., 72., 72., 72., 72., 72., 72., 72., 72. , 72. , 72. , 71.8, 71.6, 71.4, 71.2, 71. , 70.8, 70.6, 70.4, 70.2, 70. , 70. , 70. , 70. , 70. , 70. , 70. , 70. , 70. , 70. , 70. , 70. , 70. , 70.2, 70.4, 70.6, 70.8, 71. , 71.2, 71.4, 71.6, 71.8, 72.] self.assertEqual(list(lat), test_x) self.assertEqual(list(lon), test_y)
def test_get_border(self): dom = Domain(4326, "-te 25 70 35 72 -ts 500 500") result = dom.get_border(n_points=10) lat, lon = result self.assertEqual(type(lat), np.ndarray) self.assertEqual(type(lon), np.ndarray) self.assertIsInstance(result, tuple) self.assertEqual(len(result), 2) test_x = [25., 26., 27., 28., 29., 30., 31., 32., 33., 34., 35., 35., 35., 35., 35., 35., 35., 35., 35., 35., 35., 35., 35., 34., 33., 32., 31., 30., 29., 28., 27., 26., 25., 25., 25., 25., 25., 25., 25., 25., 25., 25., 25., 25.] test_y = [72., 72., 72., 72., 72., 72., 72., 72., 72., 72. , 72. , 72. , 71.8, 71.6, 71.4, 71.2, 71. , 70.8, 70.6, 70.4, 70.2, 70. , 70. , 70. , 70. , 70. , 70. , 70. , 70. , 70. , 70. , 70. , 70. , 70. , 70.2, 70.4, 70.6, 70.8, 71. , 71.2, 71.4, 71.6, 71.8, 72.] self.assertEqual(list(lat), test_x) self.assertEqual(list(lon), test_y)