示例#1
0
 def test_multiple_unknown(self):
     with self.assertRaises(ValueError):
         internal.infer_unknown_dimension((-1, 1, -1), 10)
示例#2
0
 def test_infer(self):
     assert internal.infer_unknown_dimension((-1, 2, 3), 12) == [2, 2, 3]
示例#3
0
 def test_known_all(self):
     assert internal.infer_unknown_dimension((1, 2, 3), 6) == [1, 2, 3]