use explicit ArrayDataset64 and CSRDataset64

Joan Massich · Joan Massich · commit 151fb1cee25b · 2019-02-25T15:29:40.000+01:00
diff --git a/sklearn/utils/tests/test_seq_dataset.py b/sklearn/utils/tests/test_seq_dataset.py
@@ -6,8 +6,8 @@
 from numpy.testing import assert_array_equal
 import scipy.sparse as sp
 
-from sklearn.utils.seq_dataset import ArrayDataset64 as ArrayDataset
-from sklearn.utils.seq_dataset import CSRDataset64 as CSRDataset
+from sklearn.utils.seq_dataset import ArrayDataset64
+from sklearn.utils.seq_dataset import CSRDataset64
 from sklearn.utils.seq_dataset import ArrayDataset32
 
 from sklearn.datasets import load_iris
@@ -35,8 +35,8 @@ def assert_csr_equal(X, Y):
 
 
 def test_seq_dataset():
-    dataset1 = ArrayDataset(X, y, sample_weight, seed=42)
-    dataset2 = CSRDataset(X_csr.data, X_csr.indptr, X_csr.indices,
+    dataset1 = ArrayDataset64(X, y, sample_weight, seed=42)
+    dataset2 = CSRDataset64(X_csr.data, X_csr.indptr, X_csr.indices,
                           y, sample_weight, seed=42)
 
     for dataset in (dataset1, dataset2):
@@ -59,8 +59,8 @@ def test_seq_dataset():
 
 
 def test_seq_dataset_shuffle():
-    dataset1 = ArrayDataset(X, y, sample_weight, seed=42)
-    dataset2 = CSRDataset(X_csr.data, X_csr.indptr, X_csr.indices,
+    dataset1 = ArrayDataset64(X, y, sample_weight, seed=42)
+    dataset2 = CSRDataset64(X_csr.data, X_csr.indptr, X_csr.indices,
                           y, sample_weight, seed=42)
 
     # not shuffled
@@ -91,7 +91,7 @@ def test_seq_dataset_shuffle():
 
 def test_fused_types_consistency():
     dataset32 = ArrayDataset32(X32, y32, sample_weight32, seed=42)
-    dataset64 = ArrayDataset(X, y, sample_weight, seed=42)
+    dataset64 = ArrayDataset64(X, y, sample_weight, seed=42)
 
     for i in range(5):
         # next sample
