First implementation of tests

TheMariday · Feb 17, 2024 · 93bc146 · 93bc146
1 parent 9ddaf0d
commit 93bc146
Show file tree

Hide file tree

Showing 30 changed files with 52 additions and 7 deletions.
diff --git a/.github/workflows/l3d_workflow.yml b/.github/workflows/l3d_workflow.yml
@@ -11,20 +11,21 @@ permissions:
 
 jobs:
   build:
-
-    runs-on: ubuntu-latest
+    runs-on: ${{ matrix.os }}
+    strategy:
+      matrix:
+        os: [ubuntu-latest, macos-latest, windows-latest]
+        python-version: ["3.9", "3.10", "3.11", "3.12"]
 
     steps:
     - uses: actions/checkout@v3
-    - name: Set up Python 3.10
+    - name: Set up Python
       uses: actions/setup-python@v3
-      with:
-        python-version: "3.10"
 
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install flake8
+        pip install flake8 pytest
         pip install -r requirements.txt
 
     - name: Flake 8 Syntax Errors
@@ -34,3 +35,7 @@ jobs:
     - name: Flake 8 Syntax Warnings
       run: | 
         flake8 . --count --statistics
+
+    - name: Pytest
+      run: |
+        pytest .
diff --git a/lib/led_identifier.py b/lib/led_identifier.py
@@ -16,7 +16,7 @@ def v(self):
 
 class LedFinder:
 
-    def __init__(self, threshold):
+    def __init__(self, threshold=128):
         self.threshold = threshold
 
     def find_led(self, image):

diff --git a/scripts/reconstruct.py b/scripts/reconstruct.py
@@ -2,6 +2,7 @@
 import argparse
 import os
 
+
 if __name__ == "__main__":
 
     raise NotImplementedError("This hasn't been properly implemented yet and is still a work in progress")

diff --git a/test/media/capture_sequence/a_0.png b/test/media/capture_sequence/a_0.png
diff --git a/test/media/capture_sequence/a_1.png b/test/media/capture_sequence/a_1.png
diff --git a/test/media/capture_sequence/a_10.png b/test/media/capture_sequence/a_10.png
diff --git a/test/media/capture_sequence/a_11.png b/test/media/capture_sequence/a_11.png
diff --git a/test/media/capture_sequence/a_12.png b/test/media/capture_sequence/a_12.png
diff --git a/test/media/capture_sequence/a_13.png b/test/media/capture_sequence/a_13.png
diff --git a/test/media/capture_sequence/a_14.png b/test/media/capture_sequence/a_14.png
diff --git a/test/media/capture_sequence/a_2.png b/test/media/capture_sequence/a_2.png
diff --git a/test/media/capture_sequence/a_3.png b/test/media/capture_sequence/a_3.png
diff --git a/test/media/capture_sequence/a_4.png b/test/media/capture_sequence/a_4.png
diff --git a/test/media/capture_sequence/a_5.png b/test/media/capture_sequence/a_5.png
diff --git a/test/media/capture_sequence/a_6.png b/test/media/capture_sequence/a_6.png
diff --git a/test/media/capture_sequence/a_7.png b/test/media/capture_sequence/a_7.png
diff --git a/test/media/capture_sequence/a_8.png b/test/media/capture_sequence/a_8.png
diff --git a/test/media/capture_sequence/a_9.png b/test/media/capture_sequence/a_9.png
diff --git a/test/media/capture_sequence/a_none.png b/test/media/capture_sequence/a_none.png
diff --git a/test/media/capture_sequence/s_0.png b/test/media/capture_sequence/s_0.png
diff --git a/test/media/capture_sequence/s_1.png b/test/media/capture_sequence/s_1.png
diff --git a/test/media/capture_sequence/s_2.png b/test/media/capture_sequence/s_2.png
diff --git a/test/media/capture_sequence/s_3.png b/test/media/capture_sequence/s_3.png
diff --git a/test/media/capture_sequence/s_4.png b/test/media/capture_sequence/s_4.png
diff --git a/test/media/capture_sequence/s_5.png b/test/media/capture_sequence/s_5.png
diff --git a/test/media/capture_sequence/s_6.png b/test/media/capture_sequence/s_6.png
diff --git a/test/media/capture_sequence/s_7.png b/test/media/capture_sequence/s_7.png
diff --git a/test/media/capture_sequence/s_8.png b/test/media/capture_sequence/s_8.png
diff --git a/test/media/capture_sequence/s_none.png b/test/media/capture_sequence/s_none.png
diff --git a/test/test_led_identifier.py b/test/test_led_identifier.py
@@ -0,0 +1,39 @@
+import sys
+import cv2
+sys.path.append('./')
+from lib.led_identifier import LedFinder
+
+
+def close(x, y):
+    return abs(x - y) < 0.5
+
+
+def load_image(filename):
+    image = cv2.imread(filename)
+    return cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+
+
+def test_init():
+    LedFinder()
+
+
+def test_basic_image_loading():
+    led_finder = LedFinder()
+
+    image = load_image("test/media/capture_sequence/a_0.png")
+
+    led_results = led_finder.find_led(image)
+
+    assert close(led_results.u(), 193)
+    assert close(led_results.v(), 150)
+
+
+def test_none_found():
+
+    led_finder = LedFinder()
+
+    image = load_image("test/media/capture_sequence/a_none.png")
+
+    led_results = led_finder.find_led(image)
+
+    assert led_results is None