def test_lcth_algorithm_interpolate(self):
lcthalgo = LowCloudHeightAlgorithm(**self.testinput)
cth = np.random.random_integers(0, 10, (5, 5)).astype(float)
cth[cth > 7] = np.nan
mask = np.random.random_integers(0, 1, (5, 5)).astype(bool)
result = lcthalgo.interpol_cth(cth, mask)
self.assertEqual(result.shape, (5, 5))
self.assertTrue(np.all(np.isnan(result[mask])))
self.assertGreaterEqual(np.sum(np.isnan(result)), np.sum(mask))
def test_lcth_algorithm_interpolate(self):
lcthalgo = LowCloudHeightAlgorithm(**self.testinput)
# originally obtained with np.random.random_integers
cth = np.array([[6., 3., 10., 7., 4.], [6., 9., 2., 6., 10.],
[10., 7., 4., 3., 7.], [7., 2., 5., 4., 1.],
[7., 5., 1., 4., 0.]])
cth[cth > 7] = np.nan
mask = np.array([[True, True, True, False, True],
[False, False, False, False, False],
[True, True, True, True, True],
[False, True, True, False, True],
[False, True, False, True, True]])
result = lcthalgo.interpol_cth(cth, mask)
self.assertEqual(result.shape, (5, 5))
self.assertTrue(np.all(np.isnan(result[mask])))
self.assertGreaterEqual(np.sum(np.isnan(result)), np.sum(mask))