TST change name of test such that they are included by pytest in test_split
@@ -1097,7 +1097,7 @@ def test_train_test_split():
@ignore_warnings
-def train_test_split_pandas():
+def test_train_test_split_pandas():
# check train_test_split doesn't destroy pandas dataframe
types = [MockDataFrame]
try:
@@ -1113,7 +1113,7 @@ def train_test_split_pandas():
assert isinstance(X_test, InputFeatureType)
-def train_test_split_sparse():
+def test_train_test_split_sparse():
# check that train_test_split converts scipy sparse matrices
# to csr, as stated in the documentation
X = np.arange(100).reshape((10, 10))
@@ -1125,7 +1125,7 @@ def train_test_split_sparse():
assert isinstance(X_test, csr_matrix)
-def train_test_split_mock_pandas():
+def test_train_test_split_mock_pandas():
# X mock dataframe
X_df = MockDataFrame(X)
X_train, X_test = train_test_split(X_df)
@@ -1134,7 +1134,7 @@ def train_test_split_mock_pandas():
X_train_arr, X_test_arr = train_test_split(X_df)
-def train_test_split_list_input():
+def test_train_test_split_list_input():
# Check that when y is a list / list of string labels, it works.
X = np.ones(7)
y1 = ['1'] * 4 + ['0'] * 3
