예제 #1
0
 def test_infer(self):
     assert internal.infer_unknown_dimension((-1, 2, 3), 12) == [2, 2, 3]
예제 #2
0
 def test_known_all(self):
     assert internal.infer_unknown_dimension((1, 2, 3), 6) == [1, 2, 3]
예제 #3
0
 def test_multiple_unknown(self):
     with self.assertRaises(ValueError):
         internal.infer_unknown_dimension((-1, 1, -1), 10)
예제 #4
0
 def test_infer(self):
     self.assertEqual(internal.infer_unknown_dimension((-1, 2, 3), 12),
                      [2, 2, 3])
예제 #5
0
 def test_known_all(self):
     self.assertEqual(internal.infer_unknown_dimension((1, 2, 3), 6),
                      [1, 2, 3])
예제 #6
0
 def test_infer(self):
     self.assertEqual(internal.infer_unknown_dimension((-1, 2, 3), 12),
                      [2, 2, 3])
예제 #7
0
 def test_multiple_unknown(self):
     with self.assertRaises(ValueError):
         internal.infer_unknown_dimension((-1, 1, -1), 10)
예제 #8
0
 def test_known_all(self):
     self.assertEqual(internal.infer_unknown_dimension((1, 2, 3), 6),
                      [1, 2, 3])