Skip to content
Snippets Groups Projects
Commit 756787ef authored by Denis Arrivault's avatar Denis Arrivault
Browse files

Correct tests up

parent 488e8abb
Branches
No related tags found
No related merge requests found
Pipeline #
Show whitespace changes
Inline Side-by-side
splearn/tests/test_spectral.py
+ 2
− 2
View file @ 756787ef
...@@ -49,7 +49,7 @@ class SpectralTest(unittest.TestCase): ...@@ -49,7 +49,7 @@ class SpectralTest(unittest.TestCase):
def test_version(self): def test_version(self):
adr = get_dataset_path("essai") adr = get_dataset_path("essai")
data = load_data_sample(adr=adr) 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 X = data.data
cl.fit(X=X) cl.fit(X=X)
np.testing.assert_almost_equal(cl.automaton.initial, np.testing.assert_almost_equal(cl.automaton.initial,
...@@ -57,7 +57,7 @@ class SpectralTest(unittest.TestCase): ...@@ -57,7 +57,7 @@ class SpectralTest(unittest.TestCase):
1.30974719e-02, -1.66533454e-16, 1.30974719e-02, -1.66533454e-16,
-3.70074342e-17])) -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) cl.fit(X=X)
np.testing.assert_almost_equal(cl.automaton.initial, np.testing.assert_almost_equal(cl.automaton.initial,
np.array( np.array(
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment