def test_flatten_nav(): s = Slice( origin=(0, 0, 0, 0), shape=Shape((1, 1, 1, 1), sig_dims=2) ) sflat = Slice( origin=(0, 0, 0), shape=Shape((1, 1, 1), sig_dims=2) ) assert s.flatten_nav((1, 1, 1, 1)) == sflat
def test_flatten_nav_2(): s = Slice( origin=(0, 0, 0, 0), shape=Shape((2, 16, 16, 16), sig_dims=2) ) sflat = Slice( origin=(0, 0, 0), shape=Shape((32, 16, 16), sig_dims=2) ) assert s.flatten_nav((16, 16, 16, 16)) == sflat
def test_flatten_roundtrip(): s = Slice(origin=(0, 0, 0, 0), shape=Shape((2, 16, 16, 16), sig_dims=2)) sflat = Slice(origin=(0, 0, 0), shape=Shape((32, 16, 16), sig_dims=2)) assert s.flatten_nav((16, 16, 16, 16)) == sflat assert unravel_nav(sflat, (16, 16, 16, 16)) == s