def test_get_features(self):
eop = EOPatch()
bands1 = np.arange(2 * 3 * 3 * 2).reshape(2, 3, 3, 2)
eop.add_feature(attr_type=FeatureType.DATA,
field='bands',
value=bands1)
self.assertEqual(eop.features[FeatureType.DATA]['bands'], (2, 3, 3, 2))
def test_add_feature(self):
bands = np.arange(2 * 3 * 3 * 2).reshape(2, 3, 3, 2)
eop = EOPatch()
eop.add_feature(attr_type=FeatureType.DATA, field='bands', value=bands)
self.assertTrue(np.array_equal(eop.data['bands'], bands),
msg="Data numpy array not stored")
def test_get_feature(self):
bands = np.arange(2 * 3 * 3 * 2).reshape(2, 3, 3, 2)
eop = EOPatch()
eop.add_feature(attr_type=FeatureType.DATA, field='bands', value=bands)
eop_bands = eop.get_feature(FeatureType.DATA, 'bands')
self.assertTrue(np.array_equal(eop_bands, bands),
msg="Data numpy array not returned properly")
def test_equals(self):
eop1 = EOPatch(
data={'bands': np.arange(2 * 3 * 3 * 2).reshape(2, 3, 3, 2)})
eop2 = EOPatch(
data={'bands': np.arange(2 * 3 * 3 * 2).reshape(2, 3, 3, 2)})
self.assertTrue(eop1 == eop2)
eop1.add_feature(FeatureType.DATA_TIMELESS,
field='dem',
value=np.arange(3 * 3 * 2).reshape(3, 3, 2))
self.assertFalse(eop1 == eop2)
def test_concatenate_leave_out_timeless_mismatched_keys(self):
eop1 = EOPatch()
mask1 = np.arange(3 * 3 * 2).reshape(3, 3, 2)
eop1.add_feature(attr_type=FeatureType.DATA_TIMELESS,
field='mask1',
value=mask1)
eop1.add_feature(attr_type=FeatureType.DATA_TIMELESS,
field='mask',
value=5 * mask1)
eop2 = EOPatch()
mask2 = np.arange(3 * 3 * 2).reshape(3, 3, 2)
eop2.add_feature(attr_type=FeatureType.DATA_TIMELESS,
field='mask2',
value=mask2)
eop2.add_feature(attr_type=FeatureType.DATA_TIMELESS,
field='mask',
value=5 * mask1) # add mask1 to eop2
eop = EOPatch.concatenate(eop1, eop2)
self.assertTrue('mask1' not in eop.data_timeless)
self.assertTrue('mask2' not in eop.data_timeless)
self.assertTrue('mask' in eop.data_timeless)
def test_save_load(self):
eop1 = EOPatch()
mask1 = np.arange(3 * 3 * 2).reshape(3, 3, 2)
eop1.add_feature(attr_type=FeatureType.DATA_TIMELESS,
field='mask1',
value=mask1)
eop1.add_feature(attr_type=FeatureType.DATA_TIMELESS,
field='mask',
value=5 * mask1)
eop1.save('./test_outputs/eop1/')
eop2 = EOPatch.load('./test_outputs/eop1')
self.assertEqual(eop1, eop2)
def test_concatenate_prohibit_key_mismatch(self):
eop1 = EOPatch()
bands1 = np.arange(2 * 3 * 3 * 2).reshape(2, 3, 3, 2)
eop1.add_feature(attr_type=FeatureType.DATA,
field='bands',
value=bands1)
eop2 = EOPatch()
bands2 = np.arange(3 * 3 * 3 * 2).reshape(3, 3, 3, 2)
eop2.add_feature(attr_type=FeatureType.DATA,
field='measurements',
value=bands2)
with self.assertRaises(ValueError):
EOPatch.concatenate(eop1, eop2)
def test_concatenate(self):
eop1 = EOPatch()
bands1 = np.arange(2 * 3 * 3 * 2).reshape(2, 3, 3, 2)
eop1.add_feature(attr_type=FeatureType.DATA,
field='bands',
value=bands1)
eop2 = EOPatch()
bands2 = np.arange(3 * 3 * 3 * 2).reshape(3, 3, 3, 2)
eop2.add_feature(attr_type=FeatureType.DATA,
field='bands',
value=bands2)
eop = EOPatch.concatenate(eop1, eop2)
self.assertTrue(np.array_equal(
eop.data['bands'], np.concatenate((bands1, bands2), axis=0)),
msg="Array mismatch")
def test_remove_feature(self):
bands = np.arange(2 * 3 * 3 * 2).reshape(2, 3, 3, 2)
eop = EOPatch()
eop.add_feature(attr_type=FeatureType.DATA, field='bands', value=bands)
eop.add_feature(attr_type=FeatureType.DATA,
field='bands_copy',
value=bands)
self.assertTrue('bands_copy' in eop.data.keys(),
msg="Data numpy array not stored")
self.assertTrue('bands_copy' in eop.features[FeatureType.DATA],
msg="Feature key not stored")
eop.remove_feature(attr_type=FeatureType.DATA, field='bands_copy')
self.assertFalse('bands_copy' in eop.data.keys(),
msg="Data numpy array not removed")
self.assertFalse('bands_copy' in eop.features[FeatureType.DATA],
msg="Feature key not removed")
self.assertTrue(
'bands' in eop.data.keys(),
msg="Data numpy array not stored after removal of other feature")
def test_temporal_indices(self):
""" Test case for computation of argmax/argmin of NDVI and another band
Cases with and without data masking are tested
"""
# EOPatch
eopatch = EOPatch()
t, h, w, c = 5, 3, 3, 2
# NDVI
ndvi_shape = (t, h, w, 1)
# VAlid data mask
valid_data = np.ones(ndvi_shape)
valid_data[0] = 0
valid_data[-1] = 0
# Fill in eopatch
eopatch.add_feature(FeatureType.DATA, 'NDVI', np.arange(np.prod(ndvi_shape)).reshape(ndvi_shape))
eopatch.add_feature(FeatureType.MASK, 'IS_DATA', np.ones(ndvi_shape))
eopatch.add_feature(FeatureType.MASK, 'VALID_DATA', valid_data)
# Task
add_ndvi = AddMaxMinTemporalIndicesTask(mask_data=False)
# Run task
new_eopatch = add_ndvi(eopatch)
# Asserts
self.assertTrue(np.array_equal(new_eopatch.data_timeless['ARGMIN_NDVI'], np.zeros((h, w, 1))))
self.assertTrue(np.array_equal(new_eopatch.data_timeless['ARGMAX_NDVI'], (t-1)*np.ones((h, w, 1))))
del add_ndvi, new_eopatch
# Repeat with valid dat amask
add_ndvi = AddMaxMinTemporalIndicesTask(mask_data=True)
new_eopatch = add_ndvi(eopatch)
# Asserts
self.assertTrue(np.array_equal(new_eopatch.data_timeless['ARGMIN_NDVI'], np.ones((h, w, 1))))
self.assertTrue(np.array_equal(new_eopatch.data_timeless['ARGMAX_NDVI'], (t-2)*np.ones((h, w, 1))))
del add_ndvi, new_eopatch, valid_data
# BANDS
bands_shape = (t, h, w, c)
eopatch.add_feature(FeatureType.DATA, 'BANDS', np.arange(np.prod(bands_shape)).reshape(bands_shape))
add_bands = AddMaxMinTemporalIndicesTask(data_field='BANDS',
data_index=1,
amax_data_field='ARGMAX_B1',
amin_data_field='ARGMIN_B1',
mask_data=False)
new_eopatch = add_bands(eopatch)
self.assertTrue(np.array_equal(new_eopatch.data_timeless['ARGMIN_B1'], np.zeros((h, w, 1))))
self.assertTrue(np.array_equal(new_eopatch.data_timeless['ARGMAX_B1'], (t-1)*np.ones((h, w, 1))))
def test_concatenate_leave_out_keys_with_mismatched_value(self):
mask = np.arange(3 * 3 * 2).reshape(3, 3, 2)
eop1 = EOPatch()
eop1.add_feature(attr_type=FeatureType.DATA_TIMELESS,
field='mask',
value=mask)
eop1.add_feature(attr_type=FeatureType.DATA_TIMELESS,
field='nask',
value=3 * mask)
eop2 = EOPatch()
eop2.add_feature(attr_type=FeatureType.DATA_TIMELESS,
field='mask',
value=mask)
eop2.add_feature(attr_type=FeatureType.DATA_TIMELESS,
field='nask',
value=5 * mask)
eop = EOPatch.concatenate(eop1, eop2)
self.assertTrue('mask' in eop.data_timeless)
self.assertFalse('nask' in eop.data_timeless)
def test_ndvi_slope_indices(self):
""" Test case for computation of argmax/argmin of NDVI slope
The NDVI is a sinusoid over 0-pi over the temporal dimension
Cases with and without data masking are tested
"""
# Slope needs timestamps
timestamp = perdelta(date(2018, 3, 1), date(2018, 3, 11), timedelta(days=1))
# EOPatch
eopatch = EOPatch(timestamp=list(timestamp))
t, h, w, c = 10, 3, 3, 2
# NDVI is a sinusoid where max slope is at index 1 and min slope at index 8
ndvi_shape = (t, h, w, 1)
xx = np.zeros(ndvi_shape, np.float32)
x = np.linspace(0, np.pi, t)
xx[:, :, :, :] = x[:, None, None, None]
# Valid data mask
valid_data = np.ones(ndvi_shape, np.uint8)
valid_data[1] = 0
valid_data[-1] = 0
valid_data[4] = 0
# Fill EOPatch
eopatch.add_feature(FeatureType.DATA, 'NDVI', np.sin(xx))
eopatch.add_feature(FeatureType.MASK, 'IS_DATA', np.ones(ndvi_shape))
eopatch.add_feature(FeatureType.MASK, 'VALID_DATA', valid_data)
# Tasks
add_ndvi = AddMaxMinTemporalIndicesTask(mask_data=False)
add_ndvi_slope = AddMaxMinNDVISlopeIndicesTask(mask_data=False)
# Run
new_eopatch = add_ndvi_slope(add_ndvi(eopatch))
# Assert
self.assertTrue(np.array_equal(new_eopatch.data_timeless['ARGMIN_NDVI'], (t-1)*np.ones((h, w, 1))))
self.assertTrue(np.array_equal(new_eopatch.data_timeless['ARGMAX_NDVI'], (t//2-1)*np.ones((h, w, 1))))
self.assertTrue(np.array_equal(new_eopatch.data_timeless['ARGMIN_NDVI_SLOPE'], (t-2)*np.ones((h, w, 1))))
self.assertTrue(np.array_equal(new_eopatch.data_timeless['ARGMAX_NDVI_SLOPE'], np.ones((h, w, 1))))
del add_ndvi_slope, add_ndvi, new_eopatch
# Run on valid data only now
add_ndvi = AddMaxMinTemporalIndicesTask(mask_data=True)
add_ndvi_slope = AddMaxMinNDVISlopeIndicesTask(mask_data=True)
# Run
new_eopatch = add_ndvi_slope(add_ndvi(eopatch))
# Assert
self.assertTrue(np.array_equal(new_eopatch.data_timeless['ARGMIN_NDVI'], 0 * np.ones((h, w, 1))))
self.assertTrue(np.array_equal(new_eopatch.data_timeless['ARGMAX_NDVI'], (t // 2) * np.ones((h, w, 1))))
self.assertTrue(np.array_equal(new_eopatch.data_timeless['ARGMIN_NDVI_SLOPE'], (t - 3) * np.ones((h, w, 1))))
self.assertTrue(np.array_equal(new_eopatch.data_timeless['ARGMAX_NDVI_SLOPE'], 2 * np.ones((h, w, 1))))
def test_stf_task(self):
""" Test case for computation of spatio-temporal features
The NDVI is a sinusoid over 0-pi over the temporal dimension, while bands is an array with values equal to
the temporal index
"""
# Timestamps
timestamp = perdelta(date(2018, 3, 1), date(2018, 3, 11), timedelta(days=1))
# EOPatch
eopatch = EOPatch(timestamp=list(timestamp))
# Shape of arrays
t, h, w, c = 10, 3, 3, 2
# NDVI is a sinusoid where max slope is at index 1 and min slope at index 8
ndvi_shape = (t, h, w, 1)
bands_shape = (t, h, w, c)
xx = np.zeros(ndvi_shape, np.float32)
x = np.linspace(0, np.pi, t)
xx[:, :, :, :] = x[:, None, None, None]
# Bands are arrays with values equal to the temporal index
bands = np.ones(bands_shape)*np.arange(t)[:, None, None, None]
# Add features to eopatch
eopatch.add_feature(FeatureType.DATA, 'NDVI', np.sin(xx))
eopatch.add_feature(FeatureType.DATA, 'BANDS', bands)
eopatch.add_feature(FeatureType.MASK, 'IS_DATA', np.ones(ndvi_shape))
# Tasks
add_ndvi = AddMaxMinTemporalIndicesTask(mask_data=False)
add_bands = AddMaxMinTemporalIndicesTask(data_field='BANDS',
data_index=1,
amax_data_field='ARGMAX_B1',
amin_data_field='ARGMIN_B1',
mask_data=False)
add_ndvi_slope = AddMaxMinNDVISlopeIndicesTask(mask_data=False)
add_stf = AddSpatioTemporalFeaturesTask(argmax_red='ARGMAX_B1', data_field='BANDS', indices=[0, 1])
# Run tasks
new_eopatch = add_stf(add_ndvi_slope(add_bands(add_ndvi(eopatch))))
# Asserts
self.assertTrue(new_eopatch.data_timeless['STF'].shape == (h, w, c*5))
self.assertTrue(np.array_equal(new_eopatch.data_timeless['STF'][:, :, 0:c], 4*np.ones((h, w, c))))
self.assertTrue(np.array_equal(new_eopatch.data_timeless['STF'][:, :, c:2*c], 9*np.ones((h, w, c))))
self.assertTrue(np.array_equal(new_eopatch.data_timeless['STF'][:, :, 2*c:3*c], np.ones((h, w, c))))
self.assertTrue(np.array_equal(new_eopatch.data_timeless['STF'][:, :, 3*c:4*c], 8*np.ones((h, w, c))))
self.assertTrue(np.array_equal(new_eopatch.data_timeless['STF'][:, :, 4*c:5*c], 9*np.ones((h, w, c))))
def test_check_dims(self):
bands_2d = np.arange(3 * 3).reshape(3, 3)
bands_3d = np.arange(3 * 3 * 3).reshape(3, 3, 3)
with self.assertRaises(ValueError):
EOPatch(data={'bands': bands_2d})
eop = EOPatch()
with self.assertRaises(ValueError):
eop.add_feature(attr_type=FeatureType.DATA,
field='bands',
value=bands_2d)
with self.assertRaises(ValueError):
eop.add_feature(attr_type=FeatureType.MASK,
field='mask',
value=bands_2d)
with self.assertRaises(ValueError):
eop.add_feature(attr_type=FeatureType.DATA_TIMELESS,
field='bands_timeless',
value=bands_2d)
with self.assertRaises(ValueError):
eop.add_feature(attr_type=FeatureType.MASK_TIMELESS,
field='mask_timeless',
value=bands_2d)
with self.assertRaises(ValueError):
eop.add_feature(attr_type=FeatureType.LABEL,
field='label',
value=bands_3d)
with self.assertRaises(ValueError):
eop.add_feature(attr_type=FeatureType.SCALAR,
field='scalar',
value=bands_3d)
with self.assertRaises(ValueError):
eop.add_feature(attr_type=FeatureType.LABEL_TIMELESS,
field='label_timeless',
value=bands_2d)
with self.assertRaises(ValueError):
eop.add_feature(attr_type=FeatureType.SCALAR_TIMELESS,
field='scalar_timeless',
value=bands_2d)