def test_normalize_ellipses_2_ellipses2(self):
ds1 = dshape('A, ..., B, int32')
ds2 = dshape('M, N, ..., S, T, float32')
res1, res2 = normalize_ellipses(ds1, ds2)
self.assertEqual(str(res1), 'A, N, ..., S, B, int32')
self.assertEqual(str(res2), 'M, N, ..., S, T, float32')
def test_normalize_ellipses5(self):
ds1 = dshape('..., A, B, int32')
ds2 = dshape('..., X, Y, float32')
res1, res2 = normalize_ellipses(ds1, ds2)
self.assertEqual(str(res1), '..., A, B, int32')
self.assertEqual(str(res2), '..., X, Y, float32')
def test_normalize_ellipses_2_ellipses(self):
ds1 = dshape('..., A, int32')
ds2 = dshape('X, ..., Y, Z, float32')
res1, res2 = normalize_ellipses(ds1, ds2)
self.assertEqual(str(res1), 'X, ..., Y, A, int32')
self.assertEqual(str(res2), 'X, ..., Y, Z, float32')
def test_normalize_ellipses1(self):
ds1 = dshape('..., T')
ds2 = dshape('X, Y, T')
res1, res2 = normalize_ellipses(ds1, ds2)
self.assertEqual(str(res1), 'X, Y, T')
self.assertEqual(str(res2), 'X, Y, T')
