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)
