Merge pull request #49 from ThijsIllimited/master

DDIC testing

Author: jankoslavic

examples/Showcas_LK_1D.ipynb

@@ -38,11 +38,11 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
-    "filename = 'data/data_synthetic.cih'\n",
+    "filename = '../data/data_synthetic.cih'\n",
     "video = pyidi.VideoReader(filename)"
    ]
   },
@@ -428,7 +428,7 @@
  "metadata": {
   "hide_input": false,
   "kernelspec": {
-   "display_name": ".venv",
+   "display_name": "base",
    "language": "python",
    "name": "python3"
   },
@@ -442,7 +442,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.8"
+   "version": "3.12.3"
   },
   "toc": {
    "base_numbering": 1,

tests/test_basic.py

@@ -9,6 +9,7 @@ def test_instance():
     video = pyidi.VideoReader(input_file='./data/data_synthetic.cih')
     lk = pyidi.LucasKanade(video)
     sof = pyidi.SimplifiedOpticalFlow(video)
+    lk1d = pyidi.DirectionalLucasKanade(video)
 
     assert True
     # print('test_instance: passed')
@@ -32,6 +33,15 @@ def test_points_lk():
     # print('test_points_lk: passed')
     return None
 
+def test_points_lk1d():
+    video = pyidi.VideoReader(input_file='./data/data_synthetic.cih')
+    lk1d = pyidi.DirectionalLucasKanade(video)
+    lk1d.set_points(points=[(0, 1), (1, 1)])
+
+    assert True
+    # print('test_points_lk: passed')
+    return None
+
 # def test_info():
 #     video = pyidi.pyIDI(input_file='./data/data_synthetic.cih')
 #     assert 'Shutter Speed(s)' in video.reader.info.keys()
@@ -44,6 +54,7 @@ def test_points_lk():
     test_instance()
     test_points_sof()
     test_points_lk()
+    test_points_lk1d()
 
 # if __name__ == '__mains__':
 #     np.testing.run_module_suite()

tests/test_get_displacements.py

@@ -24,6 +24,23 @@ def test_cih_lk():
 
     return None
 
+def test_cih_lk1d():
+    video = pyidi.VideoReader(input_file='./data/data_synthetic.cih')
+    lk1d = pyidi.DirectionalLucasKanade(video)
+
+    points = np.array([
+        [ 31,  35],
+        [ 31, 215],
+        [ 31, 126],
+        [ 95,  71],
+    ])
+    lk1d.set_points(points)
+    lk1d.configure(dij = (0.5, 0.5), int_order=1, verbose=0, show_pbar=False)
+    dsp = lk1d.get_displacements(autosave=False)
+
+    np.testing.assert_array_equal(dsp.shape, (len(points), video.N, 2))
+    return None
+
 def test_png_sof():
     video = pyidi.VideoReader(input_file='./data/data_synthetic_img_0.png')
     sof = pyidi.SimplifiedOpticalFlow(video)
@@ -44,4 +61,5 @@ def test_png_sof():
 
 if __name__ == '__main__':
     test_cih_lk()
+    test_cih_lk1d()
     test_png_sof()