Update tests

Author: nictru

tests/test_get_changes.py

@@ -24,14 +24,23 @@ def test_cox1_and_co1():
     assert len(sample_changes) == 2
 
     sample1_changes = sample_changes["sample1"]
-    assert len(sample1_changes) == 1
-    assert sample1_changes.iloc[0]["action"] == "copy"
-    assert sample1_changes.iloc[0]["symbol"] == "COX1"
-    assert sample1_changes.iloc[0]["new"] == "MT-CO1"
+    assert len(sample1_changes) == 0
 
     sample2_changes = sample_changes["sample2"]
     assert len(sample2_changes) == 1
-    assert sample2_changes.iloc[0]["action"] == "conflict"
+    assert sample2_changes.iloc[0]["action"] == "rename"
+
+
+def test_cox1_co1_and_ptgs1():
+    sample_symbols = {"sample1": ["MT-CO1"], "sample2": ["COX1", "PTGS1"]}
+    _, sample_changes = get_changes(sample_symbols)
+
+    assert len(sample_changes) == 2
+    assert len(sample_changes["sample1"]) == 0
+    assert len(sample_changes["sample2"]) == 1
+    assert sample_changes["sample2"].iloc[0]["action"] == "rename"
+    assert sample_changes["sample2"].iloc[0]["symbol"] == "COX1"
+    assert sample_changes["sample2"].iloc[0]["new"] == "MT-CO1"
 
 
 def test_single_sample():
@@ -88,14 +97,11 @@ def test_cox2_and_co2():
     assert len(sample_changes) == 2
 
     sample1_changes = sample_changes["sample1"]
-    assert len(sample1_changes) == 1
-    assert sample1_changes.iloc[0]["action"] == "copy"
-    assert sample1_changes.iloc[0]["symbol"] == "COX2"
-    assert sample1_changes.iloc[0]["new"] == "MT-CO2"
+    assert len(sample1_changes) == 0
 
     sample2_changes = sample_changes["sample2"]
     assert len(sample2_changes) == 1
-    assert sample2_changes.iloc[0]["action"] == "conflict"
+    assert sample2_changes.iloc[0]["action"] == "rename"
 
 
 def test_cox3():
@@ -114,15 +120,15 @@ def test_cox3():
     assert len(sample2_changes) == 0
 
 
-def test_cox3_and_co1():
+def test_cox3_and_co3():
     sample_symbols = {"sample1": ["COX3"], "sample2": ["MT-CO3", "COX3"]}
 
     _, sample_changes = get_changes(sample_symbols)
     assert len(sample_changes) == 2
 
     sample1_changes = sample_changes["sample1"]
     assert len(sample1_changes) == 1
-    assert sample1_changes.iloc[0]["action"] == "copy"
+    assert sample1_changes.iloc[0]["action"] == "rename"
     assert sample1_changes.iloc[0]["symbol"] == "COX3"
     assert sample1_changes.iloc[0]["new"] == "MT-CO3"