def test_check_chunks_unknown_shape(): with pytest.raises(ValueError): a = mt.random.rand(10, chunk_size=5) mt.random.shuffle(a) a = a.tiles() utils.check_chunks_unknown_shape([a], ValueError)
def testCheckChunksUnknownShape(self): with self.assertRaises(ValueError): a = mt.random.rand(10, chunk_size=5) mt.random.shuffle(a) a = a.tiles() utils.check_chunks_unknown_shape([a], ValueError)