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)