Ejemplo n.º 1
0
 def test_one(self):
     self.assertListEqual(list(match_shapes(int64_array([1]), int64_array([1]))), [1])
Ejemplo n.º 2
0
 def test_ones_different_lengths(self):
     self.assertListEqual(list(match_shapes(int64_array([1]), int64_array([1, 1, 1]))), [1, 1, 1])
Ejemplo n.º 3
0
 def test_many_to_one_with_trailing(self):
     self.assertListEqual(list(match_shapes(int64_array([2, 3, 4, 5]), int64_array([120, 1, 1]))), [2, 3, 4, 5, 1, 1])
Ejemplo n.º 4
0
 def test_equal_shapes(self):
     self.assertListEqual(list(match_shapes(int64_array([2, 3, 4, 5]), int64_array([2, 3, 4, 5]))), [2, 3, 4, 5])
Ejemplo n.º 5
0
 def test_trailing_one(self):
     self.assertListEqual(list(match_shapes(int64_array([1, 32, 64, 60, 1]), int64_array([8, 4, 64, 3, 20]))), [1, 8, 4, 64, 3, 20, 1])
Ejemplo n.º 6
0
 def test_many_to_one(self):
     self.assertListEqual(list(match_shapes(int64_array([2, 3, 4, 5]), int64_array([120]))), [2, 3, 4, 5])
Ejemplo n.º 7
0
 def test_ones_in_the_middle(self):
     self.assertListEqual(list(match_shapes(int64_array([32, 1, 2, 3, 1, 8]), int64_array([4, 2, 1, 4, 6, 1, 1, 8]))), [4, 2, 1, 4, 1, 2, 3, 1, 1, 8])
Ejemplo n.º 8
0
 def test_basic(self):
     self.assertListEqual(list(match_shapes(int64_array([1, 32, 64, 60]), int64_array([8, 4, 64, 3, 20]))), [1, 8, 4, 64, 3, 20])
Ejemplo n.º 9
0
 def test_not_matchabale_shapes(self):
     self.assertIsNone(match_shapes(int64_array([5, 7]), int64_array([7, 5])))
Ejemplo n.º 10
0
 def test_trailing_ones(self):
     self.assertListEqual(list(match_shapes(int64_array([1, 1, 10]), int64_array([1, 5, 1, 1, 2, 1]))), [1, 1, 5, 1, 1, 2, 1])
Ejemplo n.º 11
0
 def test_intersection_of_input_output_dimensions(self):  # is this test correct? Looks like yes...
     self.assertListEqual(list(match_shapes(int64_array([10, 20, 7]), int64_array([5, 4, 1, 70]))), [5, 2, 2, 1, 10, 7])