Update pytests based on latest data

Remove 2.7 test environment 2.7 has been removed since June 19
oncokb · Nov 1, 2023 · a0904fc · a0904fc
1 parent 2bee6c8
commit a0904fc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/pytest.yml b/.github/workflows/pytest.yml
@@ -32,7 +32,7 @@ jobs:
     strategy:
       matrix:
         os: [ ubuntu-latest, macos-latest ]
-        python-version: [ '2.7','3.7','3.8','3.9','3.10','3.11' ]
+        python-version: [ '3.7','3.8','3.9','3.10','3.11' ]
     steps:
     - uses: actions/checkout@v2
     - name: Set up Python ${{ matrix.python-version }}

diff --git a/test_Annotation.py b/test_Annotation.py
@@ -278,7 +278,7 @@ def test_cna():
     assert len(annotation) == NUMBER_OF_ANNOTATION_COLUMNS
     assert annotation[MUTATION_EFFECT_INDEX] == 'Gain-of-function'
     assert annotation[ONCOGENIC_INDEX] == 'Oncogenic'
-    assert annotation[HIGHEST_LEVEL_INDEX] == 'LEVEL_2'
+    assert annotation[HIGHEST_LEVEL_INDEX] == 'LEVEL_1'
 
     annotation = annotations[3]
     assert len(annotation) == NUMBER_OF_ANNOTATION_COLUMNS
@@ -304,7 +304,7 @@ def check_brca2_s1882_without_cancertype(annotation):
     assert annotation[MUTATION_EFFECT_INDEX] == 'Likely Loss-of-function'
     assert annotation[ONCOGENIC_INDEX] == 'Likely Oncogenic'
     assert annotation[HIGHEST_LEVEL_INDEX] == 'LEVEL_1'
-    assert annotation[LEVEL_1_INDEX] == 'Olaparib,Olaparib+Bevacizumab,Rucaparib,Niraparib'
+    assert annotation[LEVEL_1_INDEX] == 'Olaparib,Olaparib+Bevacizumab,Rucaparib,Olaparib+Abiraterone+Prednisone,Niraparib,Olaparib+Abiraterone+Prednisolone,Talazoparib+Enzalutamide'
     assert annotation[LEVEL_2_INDEX] == 'Olaparib,Rucaparib,Niraparib'
     assert annotation[LEVEL_3A_INDEX] == 'Olaparib,Talazoparib'