tests

miromannino · Feb 26, 2024 · 794766b · 794766b
1 parent bf38d1a
commit 794766b
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 4 deletions.
diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
@@ -13,7 +13,7 @@ jobs:
     strategy:
       matrix:
         os: [ubuntu-latest]
-        python-version: [3.9, 3.11, 3.12]
+        python-version: [3.9, 3.11]
 
     runs-on: ${{ matrix.os }}
 

diff --git a/tests/test_collapse_changes.py b/tests/test_collapse_changes.py
@@ -24,8 +24,10 @@ def test_collapse_changes_smaller_filesizes():
 
   files_collapsed = os.listdir(MOCK_REPO_PATH + '_c')
   files_collapsed = list(filter(lambda f: os.path.isfile(f), files_collapsed))
+  files_collapsed.remove("README.md")
   files_non_collapsed = os.listdir(MOCK_REPO_PATH)
   files_non_collapsed = list(filter(lambda f: os.path.isfile(f), files_non_collapsed))
+  files_non_collapsed.remove("README.md")
   assert len(files_collapsed) == len(files_non_collapsed)
 
   for file in files_collapsed:
@@ -36,5 +38,5 @@ def test_collapse_changes_smaller_filesizes():
       lines_of_code_collapsed = len(f.readlines())
     with open(os.path.join(MOCK_REPO_PATH, file), 'r') as f:
       lines_of_code_non_collapsed = len(f.readlines())
-    assert lines_of_code_collapsed <= lines_of_code_non_collapsed
+    assert lines_of_code_collapsed < lines_of_code_non_collapsed
 
diff --git a/tests/test_max_changes_per_file.py b/tests/test_max_changes_per_file.py
@@ -10,9 +10,35 @@ def test_max_changes_per_file():
   shutil.rmtree(MOCK_REPO_PATH, ignore_errors=True)
   mock_repo = git.Repo.init(MOCK_REPO_PATH)
 
+  shutil.rmtree(MOCK_REPO_PATH + '_c', ignore_errors=True)
+  mock_repo_collapsed = git.Repo.init(MOCK_REPO_PATH + '_c')
+  importer = Importer(repos, mock_repo_collapsed)
+  importer.set_max_changes_per_file(1)
+  importer.set_keep_commit_messages(True)
+  importer.import_repository()
+
+  shutil.rmtree(MOCK_REPO_PATH, ignore_errors=True)
+  mock_repo = git.Repo.init(MOCK_REPO_PATH)
   importer = Importer(repos, mock_repo)
-  importer.set_max_changes_per_file(5)
+  importer.set_max_changes_per_file(1)
   importer.set_keep_commit_messages(True)
   importer.import_repository()
 
-  # TODO next
+  files_collapsed = os.listdir(MOCK_REPO_PATH + '_c')
+  files_collapsed = list(filter(lambda f: os.path.isfile(f), files_collapsed))
+  files_collapsed.remove("README.md")
+  files_non_collapsed = os.listdir(MOCK_REPO_PATH)
+  files_non_collapsed = list(filter(lambda f: os.path.isfile(f), files_non_collapsed))
+  files_non_collapsed.remove("README.md")
+  assert len(files_collapsed) == len(files_non_collapsed)
+
+  for file in files_collapsed:
+    assert file in files_non_collapsed
+    lines_of_code_collapsed = 0
+    lines_of_code_non_collapsed = 0
+    with open(os.path.join(MOCK_REPO_PATH + '_c', file), 'r') as f:
+      lines_of_code_collapsed = len(f.readlines())
+    with open(os.path.join(MOCK_REPO_PATH, file), 'r') as f:
+      lines_of_code_non_collapsed = len(f.readlines())
+    print("Checking", file, "collapsed:", lines_of_code_collapsed, "non_collapsed:", lines_of_code_non_collapsed)
+    assert lines_of_code_collapsed < lines_of_code_non_collapsed