Pull request #89: Implement reset and load of default user set

Merge in HSW_SDK/vmbpy from bugfix/UNI-868-load-default-user-set-before-test-suite-execution to dev

* commit '9d4e3a6975f1ac639f08ccf93c2b5dc8072638ba':
  Restore default user set after test suite execution
  Implement reset and load of default user set

Author: NiklasKroeger-AlliedVision

Tests/helpers.py

@@ -26,6 +26,7 @@
 """
 import os
 import unittest
+import warnings
 
 import vmbpy
 
@@ -82,3 +83,33 @@ def calculate_acquisition_time(cam: vmbpy.Camera, num_frames: int) -> float:
     """
     fps = cam.get_feature_by_name('AcquisitionFrameRate').get()
     return num_frames / fps
+
+
+def reset_default_user_set(cam_id: str) -> None:
+    try:
+        with vmbpy.VmbSystem.get_instance() as vmb:
+            cam = vmb.get_camera_by_id(cam_id)
+            with cam:
+                try:
+                    cam.get_feature_by_name('UserSetDefault').set('Default')
+                except vmbpy.VmbFeatureError:
+                    try:
+                        cam.get_feature_by_name('UserSetDefaultSelector').set('Default')
+                    except vmbpy.VmbFeatureError:
+                        warnings.warn('Failed to reset default user set')
+    except vmbpy.VmbCameraError:
+        warnings.warn('Camera could not be found to reset the default user set')
+
+
+def load_default_user_set(cam_id: str) -> None:
+    try:
+        with vmbpy.VmbSystem.get_instance() as vmb:
+            cam = vmb.get_camera_by_id(cam_id)
+            with cam:
+                try:
+                    cam.get_feature_by_name('UserSetSelector').set('Default')
+                    cam.get_feature_by_name('UserSetLoad').run()
+                except vmbpy.VmbFeatureError:
+                    warnings.warn('Failed to load default user set')
+    except vmbpy.VmbCameraError:
+        warnings.warn('Camera could not be found to load default user set')

Tests/runner.py

@@ -32,7 +32,7 @@
 # Add local directory to search path for test module import in this script.
 sys.path.insert(0, os.path.abspath(os.path.dirname(__file__)))
 
-from helpers import VmbPyTestCase
+from helpers import VmbPyTestCase, reset_default_user_set, load_default_user_set
 
 # Add VmbPy module at the start of the search path. The tests should run against the
 # local vmbpy sources regardless of any existing installations.
@@ -203,6 +203,9 @@ def aligned_print(first, second):
         for mod in REAL_CAM_TEST_MODS:
             test_suite.addTests(_blacklist_tests(loader.loadTestsFromModule(mod), args.blacklist))
 
+    reset_default_user_set(VmbPyTestCase.get_test_camera_id())
+    load_default_user_set(VmbPyTestCase.get_test_camera_id())
     result = runner.run(test_suite)
-
+    reset_default_user_set(VmbPyTestCase.get_test_camera_id())
+    load_default_user_set(VmbPyTestCase.get_test_camera_id())
     sys.exit(0 if result.wasSuccessful() else 1)