Skip to content

Commit cf0bac8

Browse files
committed
expanded testcase
1 parent 3b107de commit cf0bac8

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

tests/get_grid_test.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,12 +9,15 @@
99

1010
@pytest.mark.parametrize("type", FIB_TYPES)
1111
@pytest.mark.parametrize("dim", FIB_DIMS)
12-
def test_001_gaus_sample_test(type, dim):
12+
def test_001_grid_basic(type, dim):
1313
if dim == 6:
1414
sampcount = 1000 # dim = 6, LVol = 10000, type = 'ClassicalFrolov' is unsupported
1515
else:
1616
sampcount = 10000
1717

1818
grid = get_uniform_grid(dim, sampcount, type)
1919

20-
assert grid.shape == (sampcount, dim)
20+
assert grid.shape == (sampcount, dim)
21+
22+
eps = 1e-3
23+
assert np.all((grid >= 0 - eps) & (grid <= 1 + eps))

0 commit comments

Comments
 (0)