def test_upsample_arrays(self):
array1 = np.ma.arange(10)
array2 = np.ma.arange(5)
upsampled_array1, upsampled_array2 = upsample_arrays([array1, array2])
ma_test.assert_array_equal(upsampled_array1, array1)
ma_test.assert_array_equal(upsampled_array2, array2.repeat(2))
# With mask.
array1[5:8] = np.ma.masked
array2[2:4] = np.ma.masked
upsampled_array1, upsampled_array2 = upsample_arrays([array1, array2])
result_array2 = array2.repeat(2)
result_array2[4:8] = np.ma.masked
ma_test.assert_array_equal(upsampled_array1, array1)
ma_test.assert_array_equal(upsampled_array2, result_array2)
def test_upsample_arrays(self):
array1 = np.bool_(True)
array2 = np.arange(5)
upsampled_array1, upsampled_array2 = upsample_arrays([array1, array2])
ma_test.assert_array_equal(upsampled_array1, np.array([array1] * 5))
ma_test.assert_array_equal(upsampled_array2, array2)
array1 = np.ma.arange(10)
array2 = np.ma.arange(5)
upsampled_array1, upsampled_array2 = upsample_arrays([array1, array2])
ma_test.assert_array_equal(upsampled_array1, array1)
ma_test.assert_array_equal(upsampled_array2, array2.repeat(2))
# With mask.
array1[5:8] = np.ma.masked
array2[2:4] = np.ma.masked
upsampled_array1, upsampled_array2 = upsample_arrays([array1, array2])
result_array2 = array2.repeat(2)
result_array2[4:8] = np.ma.masked
ma_test.assert_array_equal(upsampled_array1, array1)
ma_test.assert_array_equal(upsampled_array2, result_array2)
def test_upsample_arrays(self):
array1 = np.bool_(True)
array2 = np.arange(5)
upsampled_array1, upsampled_array2 = upsample_arrays([array1,
array2])
assert_array_equal(upsampled_array1, np.array([array1] * 5))
assert_array_equal(upsampled_array2, array2)
array1 = np.ma.arange(10)
array2 = np.ma.arange(5)
upsampled_array1, upsampled_array2 = upsample_arrays([array1,
array2])
assert_array_equal(upsampled_array1, array1)
assert_array_equal(upsampled_array2, array2.repeat(2))
# With mask.
array1[5:8] = np.ma.masked
array2[2:4] = np.ma.masked
upsampled_array1, upsampled_array2 = upsample_arrays([array1,
array2])
result_array2 = array2.repeat(2)
result_array2[4:8] = np.ma.masked
assert_array_equal(upsampled_array1, array1)
assert_array_equal(upsampled_array2, result_array2)
