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