Skip to content

Commit 641d0a7

Browse files
veneamueller
authored andcommitted
Cannot easily ensure non-uniqueness without the fix, just test uniqueness
1 parent 5aed364 commit 641d0a7

File tree

1 file changed

+2
-8
lines changed

1 file changed

+2
-8
lines changed

sklearn/utils/tests/test_extmath.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@
1616
from sklearn.utils.extmath import density
1717
from sklearn.utils.extmath import logsumexp
1818
from sklearn.utils.extmath import randomized_svd
19-
from sklearn.utils.extmath import svd_flip
2019
from sklearn.utils.extmath import weighted_mode
2120
from sklearn.utils.extmath import cartesian
2221
from sklearn.datasets.samples_generator import make_low_rank_matrix
@@ -244,13 +243,8 @@ def test_cartesian():
244243

245244
def test_randomized_svd_sign_flip():
246245
a = np.array([[2.0, 0.0], [0.0, 1.0]])
247-
mismatch = False # At least one pair should lead to mismatch
248246
u1, s1, v1 = randomized_svd(a, 2, flip_sign=True, random_state=41)
249-
for seed in xrange(100):
250-
u2, s2, v2 = randomized_svd(a, 2, flip_sign=False, random_state=seed)
251-
if np.any(np.sign(u1) != np.sign(u2)):
252-
mismatch = True
253-
u2, s2, v2 = svd_flip(u2, s2, v2)
247+
for seed in xrange(10):
248+
u2, s2, v2 = randomized_svd(a, 2, flip_sign=True, random_state=seed)
254249
assert_almost_equal(u1, u2)
255250
assert_almost_equal(v1, v2)
256-
assert_true(mismatch)

0 commit comments

Comments
 (0)