update test to account for mds stochasticity

maxibor · Oct 29, 2019 · 6f3fa07 · 6f3fa07
1 parent 4e3cb9e
commit 6f3fa07
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_ml.py b/tests/test_ml.py
@@ -144,7 +144,8 @@ def test_sourcemap_embed_MDS(sm):
         'a05d18d0d35a39ec3cd63464089fb2bf', 'f876bd2095f36beb3ec8925558148a15']
     assert generate_pd_md5(sm.ref_t) in [
         '96eb677be04de2db784048428b1b581d', '9cdf5d5916cc94000e19caed1d1948fd']
-    assert generate_pd_md5(sm.sink_t) == '9f7a4b87e58194db1cd7357a031de217'
+    assert generate_pd_md5(sm.sink_t) in [
+        '9f7a4b87e58194db1cd7357a031de217', '7bd707349035d92ccae7921747db63cd']
 
 
 def test_sourcemap_knn(sm):