seeds for test_mcdocker

learningmatter-mit · May 13, 2022 · 23c2495 · 23c2495
1 parent 28713e0
commit 23c2495
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/VOID/dockers/tests/test_mcdocker.py b/VOID/dockers/tests/test_mcdocker.py
@@ -16,7 +16,7 @@ def setUp(self):
         self.complex = Complex(self.host, self.guest)
         self.fitness = MinDistanceFitness(threshold=1.5)
 
-        self.num_steps = 100
+        self.num_steps = 200
         self.temperature = 0.1
 
         self.mcdocker = MonteCarloDocker(
@@ -25,10 +25,12 @@ def setUp(self):
         )
 
     def test_run(self):
+        np.random.seed(28)
         cpx = self.mcdocker.run(self.complex.copy(), self.num_steps)
         self.assertTrue(cpx.distance_matrix.min() > 1.5)
 
     def test_dock(self):
+        np.random.seed(7)
         cpxs = self.mcdocker.dock(self.num_steps)
         self.assertIsInstance(cpxs, list)
         self.assertIsInstance(cpxs[0], Complex)