Match selector to other updated tests

GMOD · Sep 20, 2024 · 14a2ada · 14a2ada
1 parent db455a3
commit 14a2ada
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/packages/apollo-cli/test/test.py b/packages/apollo-cli/test/test.py
@@ -886,8 +886,16 @@ def testFeatureChecksIndexed(self):
 
         p = shell(f"{apollo} feature get {P} -a v1")
         ff = json.loads(p.stdout)
-        g1 = [x for x in ff if x["gffId"] == "MyGene"][0]
-        g2 = [x for x in ff if x["gffId"] == "AnotherGene"][0]
+        g1 = [
+            x
+            for x in ff
+            if x["type"] == "gene" and x["attributes"]["gff_id"] == ["MyGene"]
+        ][0]
+        g2 = [
+            x
+            for x in ff
+            if x["type"] == "gene" and x["attributes"]["gff_id"] == ["AnotherGene"]
+        ][0]
 
         shell(f"{apollo} feature edit-coords {P} -i {g1['_id']} -e 201")
         shell(f"{apollo} feature edit-coords {P} -i {g2['_id']} -e 251")