コード例 #1
0
ファイル: test_internal.py プロジェクト: takagi/cupy 
 def test_multiple_unknown(self):
     with self.assertRaises(ValueError):
         internal.infer_unknown_dimension((-1, 1, -1), 10)
コード例 #2
0
ファイル: test_internal.py プロジェクト: takagi/cupy 
 def test_infer(self):
     assert internal.infer_unknown_dimension((-1, 2, 3), 12) == [2, 2, 3]
コード例 #3
0
ファイル: test_internal.py プロジェクト: takagi/cupy 
 def test_known_all(self):
     assert internal.infer_unknown_dimension((1, 2, 3), 6) == [1, 2, 3]