diff --git a/splearn/tests/test_spectral.py b/splearn/tests/test_spectral.py index 23402c11c5e7b72b277c82a5c38efc1ed523a956..00ade3452c4d7c86e6c77d04f2bca3e450736acc 100644 --- a/splearn/tests/test_spectral.py +++ b/splearn/tests/test_spectral.py @@ -49,7 +49,7 @@ class SpectralTest(unittest.TestCase): def test_version(self): adr = get_dataset_path("essai") data = load_data_sample(adr=adr) - cl = Spectral(partial=False, sparse=False, full_svd_calculation= "True", version="prefix") + cl = Spectral(partial=False, sparse=False, full_svd_calculation= True, version="prefix") X = data.data cl.fit(X=X) np.testing.assert_almost_equal(cl.automaton.initial, @@ -57,7 +57,7 @@ class SpectralTest(unittest.TestCase): 1.30974719e-02, -1.66533454e-16, -3.70074342e-17])) - cl = Spectral(partial=False, sparse=False, full_svd_calculation= "True", version="suffix") + cl = Spectral(partial=False, sparse=False, full_svd_calculation= True, version="suffix") cl.fit(X=X) np.testing.assert_almost_equal(cl.automaton.initial, np.array(