def test_create_grid(self):
with self.assertRaisesRegex(TypeError, ""):
create_grid(None)
with self.assertRaisesRegex(TypeError, ""):
create_grid((1, 1), spacing=2.0)
with self.assertRaisesRegex(TypeError, ""):
create_grid((1, 1), spacing=2.0)
test_assert(create_grid, ((1, 1), ),
np.array([[[0.0]], [[0.0]], [[1.0]]]))
test_assert(create_grid, ((1, 1), None, False),
np.array([[[0.0]], [[0.0]]]))
test_assert(create_grid, ((1, 1), (1.2, 1.3)),
np.array([[[0.0]], [[0.0]], [[1.0]]]))
test_assert(create_grid, ((1, 1, 1), (1.2, 1.3, 1.0)),
np.array([[[[0.0]]], [[[0.0]]], [[[0.0]]], [[[1.0]]]]))
test_assert(create_grid, ((1, 1, 1), (1.2, 1.3, 1.0), False),
np.array([[[[0.0]]], [[[0.0]]], [[[0.0]]]]))
g = create_grid((1, 1, 1), spacing=(1.2, 1.3, 1.0), dtype=np.int32)
np.testing.assert_equal(g.dtype, np.int32)
g = create_grid((1, 1, 1),
spacing=(1.2, 1.3, 1.0),
dtype=torch.float64,
backend="torch")
np.testing.assert_equal(g.dtype, torch.float64)
test_assert(
create_grid,
((2, 2, 2), ),
np.array([
[[[-0.5, -0.5], [-0.5, -0.5]], [[0.5, 0.5], [0.5, 0.5]]],
[[[-0.5, -0.5], [0.5, 0.5]], [[-0.5, -0.5], [0.5, 0.5]]],
[[[-0.5, 0.5], [-0.5, 0.5]], [[-0.5, 0.5], [-0.5, 0.5]]],
[[[1.0, 1.0], [1.0, 1.0]], [[1.0, 1.0], [1.0, 1.0]]],
]),
)
test_assert(
create_grid,
((2, 2, 2), (1.2, 1.3, 1.0)),
np.array([
[[[-0.6, -0.6], [-0.6, -0.6]], [[0.6, 0.6], [0.6, 0.6]]],
[[[-0.65, -0.65], [0.65, 0.65]], [[-0.65, -0.65], [0.65,
0.65]]],
[[[-0.5, 0.5], [-0.5, 0.5]], [[-0.5, 0.5], [-0.5, 0.5]]],
[[[1.0, 1.0], [1.0, 1.0]], [[1.0, 1.0], [1.0, 1.0]]],
]),
)
def test_create_grid(self):
with self.assertRaisesRegex(TypeError, ""):
create_grid(None)
with self.assertRaisesRegex(TypeError, ""):
create_grid((1, 1), spacing=2.0)
with self.assertRaisesRegex(TypeError, ""):
create_grid((1, 1), spacing=2.0)
g = create_grid((1, 1))
expected = np.array([[[0.0]], [[0.0]], [[1.0]]])
np.testing.assert_allclose(g, expected)
g = create_grid((1, 1), homogeneous=False)
expected = np.array([[[0.0]], [[0.0]]])
np.testing.assert_allclose(g, expected)
g = create_grid((1, 1), spacing=(1.2, 1.3))
expected = np.array([[[0.0]], [[0.0]], [[1.0]]])
np.testing.assert_allclose(g, expected)
g = create_grid((1, 1, 1), spacing=(1.2, 1.3, 1.0))
expected = np.array([[[[0.0]]], [[[0.0]]], [[[0.0]]], [[[1.0]]]])
np.testing.assert_allclose(g, expected)
g = create_grid((1, 1, 1), spacing=(1.2, 1.3, 1.0), homogeneous=False)
expected = np.array([[[[0.0]]], [[[0.0]]], [[[0.0]]]])
np.testing.assert_allclose(g, expected)
g = create_grid((1, 1, 1), spacing=(1.2, 1.3, 1.0), dtype=np.int32)
np.testing.assert_equal(g.dtype, np.int32)
g = create_grid((2, 2, 2))
expected = np.array([
[[[-0.5, -0.5], [-0.5, -0.5]], [[0.5, 0.5], [0.5, 0.5]]],
[[[-0.5, -0.5], [0.5, 0.5]], [[-0.5, -0.5], [0.5, 0.5]]],
[[[-0.5, 0.5], [-0.5, 0.5]], [[-0.5, 0.5], [-0.5, 0.5]]],
[[[1.0, 1.0], [1.0, 1.0]], [[1.0, 1.0], [1.0, 1.0]]],
])
np.testing.assert_allclose(g, expected)
g = create_grid((2, 2, 2), spacing=(1.2, 1.3, 1.0))
expected = np.array([
[[[-0.6, -0.6], [-0.6, -0.6]], [[0.6, 0.6], [0.6, 0.6]]],
[[[-0.65, -0.65], [0.65, 0.65]], [[-0.65, -0.65], [0.65, 0.65]]],
[[[-0.5, 0.5], [-0.5, 0.5]], [[-0.5, 0.5], [-0.5, 0.5]]],
[[[1.0, 1.0], [1.0, 1.0]], [[1.0, 1.0], [1.0, 1.0]]],
])
np.testing.assert_allclose(g, expected)