def test_normalize(): f = np.arange(64) assert len(_normalize_sequence(f, 1, 'test')) == f.ndim f = f.reshape((2,2,2,-1)) assert len(_normalize_sequence(f, 1, 'test')) == f.ndim f = f.reshape((2,2,2,1,1,-1)) assert len(_normalize_sequence(f, 1, 'test')) == f.ndim
def check(ns, val): _normalize_sequence(f.reshape(ns), val, 'test')
def test_normalize_sequence(): f = np.arange(64) assert len(_normalize_sequence(f, [1], 'test')) == f.ndim f = f.reshape((16,-1)) assert _normalize_sequence(f, [2,4], 'test') == [2,4]
def check(ns, val): with pytest.raises(ValueError): _normalize_sequence(f.reshape(ns), val, 'test')