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