def test_create_unmasked(self): rmg = RasterModelGrid((4, 5)) patches_out = rmg.patches_at_node(masked=False, nodata='bad_value') patch_values_BAD = self.patch_values.copy() patch_values_BAD[patch_values_BAD == -1] = XX assert_array_equal(patches_out, patch_values_BAD) patches_out2 = rmg.patches_at_node(masked=True, nodata=-1) assert_array_equal(patches_out2.data, self.patch_values) assert_array_equal(patches_out2.mask, self.patch_mask)
def test_create_masked(self): rmg = RasterModelGrid((4, 5)) patches_out = rmg.patches_at_node(masked=True) assert_array_equal(patches_out.data, self.patch_values) assert_array_equal(patches_out.mask, self.patch_mask) # check we can change: patches_out2 = rmg.patches_at_node(masked=False) assert_array_equal(patches_out2, patches_out.data)