Merge pull request #1326 from thempel/fix_submodel_tests

[msm tests] hard-code submodel test dtraj
markovmodel · Jun 28, 2018 · a2750f0 · a2750f0
2 parents 9deb3ae + 577d4b0
commit a2750f0
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 4 deletions.
diff --git a/pyemma/msm/tests/test_bayesian_hmsm.py b/pyemma/msm/tests/test_bayesian_hmsm.py
@@ -306,8 +306,14 @@ def test_submodel_simple(self):
         # sanity check for submodel;
         # call should not alter self
         from copy import deepcopy
-        dtrj = np.random.randint(0, 2, size=100)
-        dtrj[np.random.randint(0, dtrj.shape[0], 3)] = 2
+        # dtrj = np.random.randint(0, 2, size=100)
+        # dtrj[np.random.randint(0, dtrj.shape[0], 3)] = 2
+        # hard-coded due to stochastic failures
+        dtrj = [1, 0, 0, 1, 0, 1, 1, 0, 0, 1, 1, 1, 0, 0, 0, 0, 1, 0, 1, 1, 0, 1, 1, 0, 0, 0,
+                0, 2, 1, 0, 1, 1, 0, 1, 0, 1, 1, 1, 0, 1, 0, 1, 1, 0, 1, 1, 0, 1, 0, 0, 1, 0,
+                1, 1, 0, 1, 0, 0, 0, 1, 1, 0, 1, 1, 0, 0, 1, 1, 2, 0, 0, 1, 1, 2, 0, 1, 1, 1,
+                0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 0, 1, 1, 0, 0, 0, 1, 1, 0, 1, 1, 0]
+
         h = bayesian_hidden_markov_model(dtrj, 3, 2)
         h_original = deepcopy(h)
 

diff --git a/pyemma/msm/tests/test_hmsm.py b/pyemma/msm/tests/test_hmsm.py
@@ -440,8 +440,14 @@ def test_submodel_simple(self):
         # sanity check for submodel;
         # call should not alter self
         from copy import deepcopy
-        dtrj = np.random.randint(0, 2, size=100)
-        dtrj[np.random.randint(0, dtrj.shape[0], 3)] = 2
+        # dtrj = np.random.randint(0, 2, size=100)
+        # dtrj[np.random.randint(0, dtrj.shape[0], 3)] = 2
+        # hard-coded due to stochastic failures
+        dtrj = [1, 0, 0, 1, 0, 1, 1, 0, 0, 1, 1, 1, 0, 0, 0, 0, 1, 0, 1, 1, 0, 1, 1, 0, 0, 0,
+                0, 2, 1, 0, 1, 1, 0, 1, 0, 1, 1, 1, 0, 1, 0, 1, 1, 0, 1, 1, 0, 1, 0, 0, 1, 0,
+                1, 1, 0, 1, 0, 0, 0, 1, 1, 0, 1, 1, 0, 0, 1, 1, 2, 0, 0, 1, 1, 2, 0, 1, 1, 1,
+                0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 0, 1, 1, 0, 0, 0, 1, 1, 0, 1, 1, 0]
+
         h = msm.estimate_hidden_markov_model(dtrj, 3, 2)
         h_original = deepcopy(h)