Update tests.

broadinstitute · Aug 25, 2023 · 6bceb0e · 6bceb0e
1 parent 120c5da
commit 6bceb0e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 9 deletions.
diff --git a/reference_data/management/tests/update_gene_shet_tests.py b/reference_data/management/tests/update_gene_shet_tests.py
@@ -2,12 +2,12 @@
 from reference_data.management.tests.test_utils import ReferenceDataCommandTestCase
 
 class UpdateGeneShetTest(ReferenceDataCommandTestCase):
-    URL = 'https://storage.googleapis.com/seqr-reference-data/Shet/Shet_Zeng_2023.tsv'
+    URL = 'https://storage.googleapis.com/seqr-reference-data/gene_constraint/shet_Zeng(2023).xlsx%20-%20All%20scores-for%20gene%20page.tsv'
     DATA = [
-        'ensg	hgnc	post_mean_shet	shet_constrained\n',
-        'ENSG00000223972	HGNC:37225	3.01E-05	0\n',
-        'ENSG00000227233	HGNC:26441	4.85E-05	0\n',
-        'ENSG00000243485	HGNC:4013	5.08E-05	1\n',
+        'ensg	hgnc	post_mean (Shet)\n',
+        'ENSG00000223972	HGNC:37225	3.01E-05\n',
+        'ENSG00000227233	HGNC:26441	4.85E-05\n',
+        'ENSG00000243485	HGNC:4013	5.08E-05\n',
     ]
 
     def test_update_gene_cn_sensitivity_command(self):
@@ -16,7 +16,5 @@ def test_update_gene_cn_sensitivity_command(self):
         self.assertEqual(GeneShet.objects.count(), 2)
         record = GeneShet.objects.get(gene__gene_id='ENSG00000223972')
         self.assertEqual(record.shet, 3.01E-05)
-        self.assertEqual(record.shet_constrained, False)
         record = GeneShet.objects.get(gene__gene_id='ENSG00000243485')
         self.assertEqual(record.shet, 5.08E-05)
-        self.assertEqual(record.shet_constrained, True)
diff --git a/seqr/fixtures/reference_data.json b/seqr/fixtures/reference_data.json
@@ -1173,8 +1173,7 @@
     "pk": 1,
     "fields": {
         "gene": 1,
-        "shet": 0.90576,
-        "shet_constrained": true
+        "shet": 0.90576
     }
 },
     {