Merge pull request #10 from eriknovak/feature/tests

Unit test book keeping and expanding test environment on all OS

.github/workflows/unittests.yaml

@@ -8,10 +8,11 @@ on:
 
 jobs:
   test:
-    runs-on: ubuntu-latest
+    runs-on: ${{ matrix.os }}
     strategy:
       matrix:
-        python-version: ["3.8", "3.9", "3.10"] # "3.11" - for some reason the models give different predictions on 3.11
+        os: [ubuntu-latest, macos-latest, windows-latest]
+        python-version: ["3.8", "3.9", "3.10"]
     steps:
       - uses: actions/checkout@v4
       - name: Set up Python ${{ matrix.python-version }}

.gitignore

@@ -134,12 +134,10 @@ dmypy.json
 
 .vscode
 
+data/*
 data/**/
 !data/README.md
 
-models/**/
-!models/README.md
-
 notebooks
 !docs/documentation/notebooks
 

test/test_extractors.py

@@ -1,4 +1,5 @@
 import unittest
+import warnings
 
 from anonipy.definitions import Entity
 from anonipy.anonymize.extractors import EntityExtractor
@@ -97,6 +98,12 @@
 
 
 class TestEntityExtractor(unittest.TestCase):
+
+    def setUp(self):
+        warnings.filterwarnings("ignore", category=ImportWarning)
+        warnings.filterwarnings("ignore", category=UserWarning)
+        warnings.filterwarnings("ignore", category=FutureWarning)
+
     def test_init(self):
         try:
             EntityExtractor()

test/test_generators.py

@@ -1,5 +1,6 @@
 import re
 import unittest
+import warnings
 
 import torch
 
@@ -124,6 +125,11 @@ class TestMaskLabelGenerator(unittest.TestCase):
     def setUpClass(self):
         self.generator = MaskLabelGenerator()
 
+    def setUp(self):
+        warnings.filterwarnings("ignore", category=ImportWarning)
+        warnings.filterwarnings("ignore", category=UserWarning)
+        warnings.filterwarnings("ignore", category=FutureWarning)
+
     def test_has_methods(self):
         self.assertEqual(hasattr(self.generator, "generate"), True)
 

test/test_language_detector.py

@@ -1,4 +1,5 @@
 import unittest
+import warnings
 
 from anonipy.utils.language_detector import LanguageDetector
 from anonipy.constants import LANGUAGES
@@ -10,6 +11,11 @@
 
 class TestLanguageDetector(unittest.TestCase):
 
+    def setUp(self):
+        warnings.filterwarnings("ignore", category=ImportWarning)
+        warnings.filterwarnings("ignore", category=UserWarning)
+        warnings.filterwarnings("ignore", category=FutureWarning)
+
     def test_init(self):
         language_detector = LanguageDetector()
         self.assertEqual(language_detector.__class__, LanguageDetector)