fix arg reduction test (#32)

mars-project · Dec 14, 2018 · e186b3c · e186b3c
1 parent 74d7936
commit e186b3c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mars/tensor/execution/tests/test_reduction_execute.py b/mars/tensor/execution/tests/test_reduction_execute.py
@@ -271,8 +271,8 @@ def testArgReduction(self):
 
         random_min = np.random.randint(0, 200)
         random_max = np.random.randint(200, 400)
-        raw[np.unravel_index(random_min, raw.shape)] = -1
-        raw[np.unravel_index(random_max, raw.shape)] = 2
+        raw_format[np.unravel_index(random_min, raw_format.shape)] = -1
+        raw_format[np.unravel_index(random_max, raw_format.shape)] = 2
 
         raw = raw_format.tocoo()
         arr = tensor(raw, chunks=3)