def test_quarter(self):
x1, y1 = make_xy(4, 1)
x2, y2 = make_xy(4, 1, quarter=True)
np.testing.assert_array_equal(x2, x1[2:4, 2:4])
np.testing.assert_array_equal(y2, y1[2:4, 2:4])
def test_odd_sampling(self):
ref = [-0.8, -0.4, 0, 0.4, 0.8]
arr1 = make_xy(5, 1, vector=True)
arr2 = make_xy(5, 1, vector=True, zero_sampled=True)
np.testing.assert_array_equal(arr1, ref)
np.testing.assert_array_equal(arr2, ref)
def test_makexy(self):
domain = DomainXY.from_makexy(5, 2)
x, y = make_xy(5, 2)
np.testing.assert_array_equal(x, domain.xmap)
np.testing.assert_array_equal(y, domain.ymap)
def test_polar(self):
x, y = make_xy(2, 1, polar=True)
ref_x = np.ones((2, 2)) * np.cos(45 * np.pi / 180.)
ref_y = np.array([[-135, -45], [135, 45]]) * np.pi / 180.
np.testing.assert_allclose(x, ref_x)
np.testing.assert_allclose(y, ref_y)
def test_xy(self):
'''Test that x and y are transposed wrt. each other'''
x, y = make_xy(2, 2)
assert np.all(x.T == y)
def test_zero_sampled(self):
ref = [-1, -0.5, 0, 0.5]
arr = make_xy(4, 1, vector=True, zero_sampled=True)
np.testing.assert_array_equal(arr, ref)
def test_vector_sampling(self):
ref = [-0.75, -0.25, 0.25, 0.75]
arr = make_xy(4, 1, vector=True)
np.testing.assert_array_equal(arr, ref)
def test_sampling_fails(self):
with self.assertRaises(ValueError):
make_xy(1, 2)
def from_makexy(cls, *args, **kwargs):
'''Same arguments as make_xy'''
x, y = make_xy(*args, **kwargs)
return cls.from_xy_maps(x, y)