def test_grid_can_have_a_single_cell(): shape = () for i in range(10): shape = shape + (1,) g = Grid(shape) assert len(g) == 1 assert g.dim() == len(shape) assert tuple(g.shape()) == shape
def test_grid_one_dim_same_as_list(): l = list(range(10)) g = Grid((len(l),)) assert len(g) == len(l) assert g.dim() == 1 assert tuple(g) == tuple(l) for i in l: assert g.coordinates(i) == (i,) assert g.index((i,)) == i
def test_grid_can_be_empty(): g = Grid(()) assert len(g) == 0 assert g.dim() == 0 assert tuple(g.shape()) == ()
def test_grid_can_be_null_along_one_axis(): g = Grid((3, 0, 4)) assert len(g) == 0 assert g.dim() == 3 assert_raises(IndexError, lambda: g.index((0, 0, 0))) assert_raises(IndexError, lambda: g.coordinates(0))