Add test for matching cv2 constants

snake-pit/test_cv2.py

@@ -7,6 +7,29 @@
 TMP_PATH = "tmp.mp4"
 
 
+def test_constants():
+    assert ocv_cv2.CAP_PROP_POS_MSEC == vf_cv2.CAP_PROP_POS_MSEC
+    assert ocv_cv2.CAP_PROP_POS_FRAMES == vf_cv2.CAP_PROP_POS_FRAMES
+    assert ocv_cv2.CAP_PROP_FRAME_WIDTH == vf_cv2.CAP_PROP_FRAME_WIDTH
+    assert ocv_cv2.CAP_PROP_FRAME_HEIGHT == vf_cv2.CAP_PROP_FRAME_HEIGHT
+    assert ocv_cv2.CAP_PROP_FPS == vf_cv2.CAP_PROP_FPS
+
+    assert ocv_cv2.FONT_HERSHEY_SIMPLEX == vf_cv2.FONT_HERSHEY_SIMPLEX
+    assert ocv_cv2.FONT_HERSHEY_PLAIN == vf_cv2.FONT_HERSHEY_PLAIN
+    assert ocv_cv2.FONT_HERSHEY_DUPLEX == vf_cv2.FONT_HERSHEY_DUPLEX
+    assert ocv_cv2.FONT_HERSHEY_COMPLEX == vf_cv2.FONT_HERSHEY_COMPLEX
+    assert ocv_cv2.FONT_HERSHEY_TRIPLEX == vf_cv2.FONT_HERSHEY_TRIPLEX
+    assert ocv_cv2.FONT_HERSHEY_COMPLEX_SMALL == vf_cv2.FONT_HERSHEY_COMPLEX_SMALL
+    assert ocv_cv2.FONT_HERSHEY_SCRIPT_SIMPLEX == vf_cv2.FONT_HERSHEY_SCRIPT_SIMPLEX
+    assert ocv_cv2.FONT_HERSHEY_SCRIPT_COMPLEX == vf_cv2.FONT_HERSHEY_SCRIPT_COMPLEX
+    assert ocv_cv2.FONT_ITALIC == vf_cv2.FONT_ITALIC
+
+    assert ocv_cv2.FILLED == vf_cv2.FILLED
+    assert ocv_cv2.LINE_4 == vf_cv2.LINE_4
+    assert ocv_cv2.LINE_8 == vf_cv2.LINE_8
+    assert ocv_cv2.LINE_AA == vf_cv2.LINE_AA
+
+
 def rw(cv2):
     cap = cv2.VideoCapture(VID_PATH)
     assert cap.isOpened()

vidformer-py/vidformer/cv2/vf_cv2.py

@@ -8,11 +8,11 @@
     bin="../target/release/vidformer-cli"
 )  # TODO: don't hardcode this
 
-CAP_PROP_FPS = "CAP_PROP_FPS"
-CAP_PROP_FRAME_WIDTH = "CAP_PROP_FRAME_WIDTH"
-CAP_PROP_FRAME_HEIGHT = "CAP_PROP_FRAME_HEIGHT"
-CAP_PROP_POS_MSEC = "CAP_PROP_POS_MSEC"
-CAP_PROP_POS_FRAMES = "CAP_PROP_POS_FRAMES"
+CAP_PROP_POS_MSEC = 0
+CAP_PROP_POS_FRAMES = 1
+CAP_PROP_FRAME_WIDTH = 3
+CAP_PROP_FRAME_HEIGHT = 4
+CAP_PROP_FPS = 5
 
 FONT_HERSHEY_SIMPLEX = 0
 FONT_HERSHEY_PLAIN = 1
@@ -24,7 +24,7 @@
 FONT_HERSHEY_SCRIPT_COMPLEX = 7
 FONT_ITALIC = 16
 
-LINE_FILLED = -1
+FILLED = -1
 LINE_4 = 4
 LINE_8 = 8
 LINE_AA = 16