def test_core_greater_equal(self):
a, b = shape_poly.parse_spec("a, b", (2, 3))
self.assertTrue(core.greater_equal_dim(a, a))
self.assertTrue(core.greater_equal_dim(a, 0))
self.assertTrue(core.greater_equal_dim(a, 1))
self.assertTrue(core.greater_equal_shape((a, 2), (1, 1)))
with self.assertRaisesRegex(core.InconclusiveDimensionOperation,
"Dimension polynomial comparison .* is inconclusive"):
core.greater_equal_dim(a, 2)
with self.assertRaisesRegex(core.InconclusiveDimensionOperation,
"Dimension polynomial comparison .* is inconclusive"):
core.greater_equal_dim(a, b)
def test_dim_vars_greater_equal(self):
da, db = shape_poly.parse_spec("a, b", (2, 3))
self.assertTrue(core.greater_equal_dim(da, da))
self.assertTrue(core.greater_equal_dim(da, 0))
self.assertTrue(core.greater_equal_dim(da, 1))
self.assertTrue(core.greater_equal_shape((da, 2), (1, 1)))
with self.assertRaisesRegex(core.InconclusiveDimensionOperation,
"Shape variable comparison .* is inconclusive"):
core.greater_equal_dim(da, 2)
with self.assertRaisesRegex(core.InconclusiveDimensionOperation,
"Shape variable comparison .* is inconclusive"):
core.greater_equal_dim(da, db)
