def test_sliced_getter():
def get_slice(slice_obj=None):
arr = np.ones((10, 10))
if slice_obj is None:
slice_obj = make_slice[:, :]
return arr[slice_obj]
sli = SlicedGetter(get_slice)
test = make_slice[2:-2, 2:-2]
assert sli[test].shape == (6, 6)
def force(self):
return SlicedGetter(self.force_slice)
def velocity(self):
return SlicedGetter(self.velocity_slice)
def mu(self):
return SlicedGetter(self.mu_slice)
def concentration(self):
return SlicedGetter(self.concentration_slice)
def phi(self):
return SlicedGetter(self.phi_slice)
def density(self):
return SlicedGetter(self.density_slice)