Fix test

UCL-ARC · Oct 11, 2024 · 64193a4 · 64193a4
1 parent a9f6951
commit 64193a4
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/test_package_generation.py b/tests/test_package_generation.py
@@ -46,7 +46,6 @@ def test_package_generation(
  # Check main files and directories inside
  expected_files: set[pathlib.Path] = {
  pathlib.Path(),
- pathlib.Path(".git"),
  pathlib.Path(".github"),
  pathlib.Path(".github/ISSUE_TEMPLATE"),
  pathlib.Path(".github/ISSUE_TEMPLATE/bug_report.yml"),
@@ -80,10 +79,11 @@ def test_package_generation(
 
  actual_files = get_all_files_folders(test_project_dir)
  # Filter out anything under specific directories to make comparison easier
- actual_files = {
- path
- for path in actual_files
- if not str(path).startswith((".git/", "__pycache__/"))
+ actual_files = actual_files - {
+ a
+ for a in actual_files
+ if len(a.parts) > 0
+ and (a.parts[0] == ".git" or "__pycache__" in a.parts or ".DS_Store" in a.parts)
  }
 
  assert actual_files == expected_files