fix: remove the entry of a test class if its list is empty (#983)

STAMP-project · Jan 5, 2021 · 1d17d36 · 1d17d36
1 parent 5de4962
commit 1d17d36
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...rc/main/java/eu/stamp_project/diff_test_selection/selector/EnhancedDiffTestSelection.java b/...rc/main/java/eu/stamp_project/diff_test_selection/selector/EnhancedDiffTestSelection.java
@@ -58,9 +58,9 @@ public Map<String, Set<String>> selectTests() {
         for (String testClassName : modifiedTestsPerTestClass.keySet()) {
             if (selectTestsPerTestClasses.containsKey(testClassName)) {
                 selectTestsPerTestClasses.get(testClassName).removeAll(modifiedTestsPerTestClass.get(testClassName));
-            }
-            if (selectTestsPerTestClasses.get(testClassName).isEmpty()) {
-                selectTestsPerTestClasses.remove(testClassName);
+                if (selectTestsPerTestClasses.get(testClassName).isEmpty()) {
+                    selectTestsPerTestClasses.remove(testClassName);
+                }
             }
         }
         return selectTestsPerTestClasses;