From 2fd837500a221a6bffa6de65591071fcd05846f8 Mon Sep 17 00:00:00 2001 From: Jan Petykiewicz Date: Sun, 31 Oct 2021 19:03:05 -0700 Subject: [PATCH] Add some tests --- gridlock/test/__init__.py | 0 gridlock/test/test_grid.py | 88 ++++++++++++++++++++++++++++++++++++++ 2 files changed, 88 insertions(+) create mode 100644 gridlock/test/__init__.py create mode 100644 gridlock/test/test_grid.py diff --git a/gridlock/test/__init__.py b/gridlock/test/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/gridlock/test/test_grid.py b/gridlock/test/test_grid.py new file mode 100644 index 0000000..fc54030 --- /dev/null +++ b/gridlock/test/test_grid.py @@ -0,0 +1,88 @@ +import pytest # type: ignore +import numpy # type: ignore +from numpy.testing import assert_allclose, assert_array_equal # type: ignore + +from .. import Grid + + +def test_draw_oncenter_2x2() -> None: + xs = [-1, 0, 1] + ys = [-1, 0, 1] + zs = [-1, 1] + grid = Grid([xs, ys, zs], shifts=[[0, 0, 0]]) + arr = grid.allocate(0) + + grid.draw_cuboid(arr, center=[0, 0, 0], dimensions=[1, 1, 10], foreground=1) + + correct = numpy.array([[0.25, 0.25], + [0.25, 0.25]])[None, :, :, None] + + assert_allclose(arr, correct) + + +def test_draw_ongrid_4x4() -> None: + xs = [-2, -1, 0, 1, 2] + ys = [-2, -1, 0, 1, 2] + zs = [-1, 1] + grid = Grid([xs, ys, zs], shifts=[[0, 0, 0]]) + arr = grid.allocate(0) + + grid.draw_cuboid(arr, center=[0, 0, 0], dimensions=[2, 2, 10], foreground=1) + + correct = numpy.array([[0, 0, 0, 0], + [0, 1, 1, 0], + [0, 1, 1, 0], + [0, 0, 0, 0]])[None, :, :, None] + + assert_allclose(arr, correct) + + +def test_draw_xshift_4x4() -> None: + xs = [-2, -1, 0, 1, 2] + ys = [-2, -1, 0, 1, 2] + zs = [-1, 1] + grid = Grid([xs, ys, zs], shifts=[[0, 0, 0]]) + arr = grid.allocate(0) + + grid.draw_cuboid(arr, center=[0.5, 0, 0], dimensions=[1.5, 2, 10], foreground=1) + + correct = numpy.array([[0, 0, 0, 0], + [0, 0.25, 0.25, 0], + [0, 1, 1, 0], + [0, 0.25, 0.25, 0]])[None, :, :, None] + + assert_allclose(arr, correct) + + +def test_draw_yshift_4x4() -> None: + xs = [-2, -1, 0, 1, 2] + ys = [-2, -1, 0, 1, 2] + zs = [-1, 1] + grid = Grid([xs, ys, zs], shifts=[[0, 0, 0]]) + arr = grid.allocate(0) + + grid.draw_cuboid(arr, center=[0, 0.5, 0], dimensions=[2, 1.5, 10], foreground=1) + + correct = numpy.array([[0, 0, 0, 0], + [0, 0.25, 1, 0.25], + [0, 0.25, 1, 0.25], + [0, 0, 0, 0]])[None, :, :, None] + + assert_allclose(arr, correct) + + +def test_draw_2shift_4x4() -> None: + xs = [-2, -1, 0, 1, 2] + ys = [-2, -1, 0, 1, 2] + zs = [-1, 1] + grid = Grid([xs, ys, zs], shifts=[[0, 0, 0]]) + arr = grid.allocate(0) + + grid.draw_cuboid(arr, center=[0.5, 0, 0], dimensions=[1.5, 1, 10], foreground=1) + + correct = numpy.array([[0, 0, 0, 0], + [0, 0.125, 0.125, 0], + [0, 0.5, 0.5, 0], + [0, 0.125, 0.125, 0]])[None, :, :, None] + + assert_allclose(arr, correct)