def test_partial_shape():
ps = PartialShape([1, 2, 3, 4])
assert ps.is_static
assert not ps.is_dynamic
assert ps.rank == 4
assert repr(ps) == "<PartialShape: {1,2,3,4}>"
shape = Shape([1, 2, 3])
ps = PartialShape(shape)
assert ps.is_static
assert not ps.is_dynamic
assert ps.all_non_negative
assert ps.rank == 3
assert list(ps.get_shape()) == [1, 2, 3]
assert list(ps.get_max_shape()) == [1, 2, 3]
assert list(ps.get_min_shape()) == [1, 2, 3]
assert list(ps.to_shape()) == [1, 2, 3]
assert repr(shape) == "<Shape{1, 2, 3}>"
assert repr(ps) == "<PartialShape: {1,2,3}>"
ps = PartialShape(
[Dimension(1),
Dimension(2),
Dimension(3),
Dimension.dynamic()])
assert not ps.is_static
assert ps.is_dynamic
assert ps.all_non_negative
assert ps.rank == 4
assert list(ps.get_min_shape()) == [1, 2, 3, 0]
assert list(ps.get_max_shape())[3] > 1000000000
assert repr(ps) == "<PartialShape: {1,2,3,?}>"
ps = PartialShape([1, 2, 3, -1])
assert not ps.is_static
assert ps.is_dynamic
assert ps.all_non_negative
assert ps.rank == 4
assert list(ps.get_min_shape()) == [1, 2, 3, 0]
assert list(ps.get_max_shape())[3] > 1000000000
assert repr(ps) == "<PartialShape: {1,2,3,?}>"
ps = PartialShape.dynamic()
assert not ps.is_static
assert ps.is_dynamic
assert ps.rank == Dimension.dynamic()
assert list(ps.get_min_shape()) == []
assert list(ps.get_max_shape()) == []
assert repr(ps) == "<PartialShape: ?>"
ps = PartialShape.dynamic(r=Dimension(2))
assert not ps.is_static
assert ps.is_dynamic
assert ps.rank == 2
assert 2 == ps.rank
assert list(ps.get_min_shape()) == [0, 0]
assert list(ps.get_max_shape())[0] > 1000000000
assert repr(ps) == "<PartialShape: {?,?}>"
def test_dimension():
dim = Dimension()
assert dim.is_dynamic
assert not dim.is_static
assert repr(dim) == "<Dimension: ?>"
dim = Dimension.dynamic()
assert dim.is_dynamic
assert not dim.is_static
assert repr(dim) == "<Dimension: ?>"
dim = Dimension(10)
assert dim.is_static
assert len(dim) == 10
assert dim.get_length() == 10
assert dim.get_min_length() == 10
assert dim.get_max_length() == 10
assert repr(dim) == "<Dimension: 10>"
dim = Dimension(5, 15)
assert dim.is_dynamic
assert dim.get_min_length() == 5
assert dim.get_max_length() == 15
assert repr(dim) == "<Dimension: [5, 15]>"
def test_dimension_comparisons():
d1 = Dimension.dynamic()
d2 = Dimension.dynamic()
assert d1 == d2
assert d1 == -1
assert d1.refines(d2)
assert d1.relaxes(d2)
assert d2.refines(d1)
assert d2.relaxes(d1)
assert d2.compatible(d1)
assert d2.same_scheme(d1)
d1 = Dimension.dynamic()
d2 = Dimension(3)
assert d1 != d2
assert d2 == 3
assert not d1.refines(d2)
assert d1.relaxes(d2)
assert d2.refines(d1)
assert not d2.relaxes(d1)
assert d2.compatible(d1)
assert not d2.same_scheme(d1)
d1 = Dimension(3)
d2 = Dimension(3)
assert d1 == d2
assert d1.refines(d2)
assert d1.relaxes(d2)
assert d2.refines(d1)
assert d2.relaxes(d1)
assert d2.compatible(d1)
assert d2.same_scheme(d1)
d1 = Dimension(4)
d2 = Dimension(3)
assert d1 != d2
assert not d1.refines(d2)
assert not d1.relaxes(d2)
assert not d2.refines(d1)
assert not d2.relaxes(d1)
assert not d2.compatible(d1)
assert not d2.same_scheme(d1)