Fixed failing tests

protectai · Feb 22, 2024 · 6491646 · 6491646
1 parent d17bc83
commit 6491646
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/tests/test_modelscan.py b/tests/test_modelscan.py
@@ -1425,18 +1425,20 @@ def test_scan_tensorflow(tensorflow_file_path: Any) -> None:
     ms = ModelScan()
     results = ms.scan(Path(f"{safe_tensorflow_model_dir}"))
     assert ms.issues.all_issues == []
-    assert results["summary"]["scanned"]["scanned_files"] == [
+    assert set(results["summary"]["scanned"]["scanned_files"]) == {
         f"fingerprint.pb",
         f"keras_metadata.pb",
         f"saved_model.pb",
-    ]
-    assert [
-        skipped_file["source"]
-        for skipped_file in results["summary"]["skipped"]["skipped_files"]
-    ] == [
+    }
+    assert set(
+        [
+            skipped_file["source"]
+            for skipped_file in results["summary"]["skipped"]["skipped_files"]
+        ]
+    ) == {
         f"variables/variables.data-00000-of-00001",
         f"variables/variables.index",
-    ]
+    }
     assert results["errors"] == []
 
     file_name = "saved_model.pb"