jan/gridlock
2
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add some tests

Browse Source
This commit is contained in:
Jan Petykiewicz 2021-10-31 19:03:05 -07:00
parent 85d9ff7ace
commit 2fd837500a
2 changed files with 88 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
gridlock/test/__init__.py Normal file
View File

88
gridlock/test/test_grid.py Normal file
View File

@ -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)