Fixed utr tests

KarrLab · Jun 20, 2018 · 8d1590c · 8d1590c
1 parent b272580
commit 8d1590c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/kb_gen/test_kb_gen_genome.py b/tests/kb_gen/test_kb_gen_genome.py
@@ -20,7 +20,7 @@ def setUp(self):
 
         # Creates the GenomeGenerator object and sets the parameters as given
         options = {
-            'mean_num_genes': 1000}
+            'mean_num_genes': 50}
         self.gen = genome.GenomeGenerator(kb, options)
         self.gen.run()
 
@@ -141,9 +141,9 @@ def test_utrs(self):
         three_prime_len = self.gen.options.get('three_prime_len')
 
         self.assertAlmostEqual(sum_five_prime/mRnaCount,
-                               five_prime_len, 3 * math.sqrt(five_prime_len))
+                               five_prime_len, delta=3 * math.sqrt(five_prime_len))
         self.assertAlmostEqual(sum_three_prime/mRnaCount,
-                               three_prime_len, 3 * math.sqrt(three_prime_len))
+                               three_prime_len, delta=3 * math.sqrt(three_prime_len))
 
     def test_operons(self):
         tus = self.gen.knowledge_base.cell.loci.get(