v0.2 to master

1 files changed, 52 insertions, 0 deletions

diff --git a/tests/test_vimba.py b/tests/test_vimba.py
new file mode 100644
index 0000000..b1ef987
--- /dev/null
+++ b/tests/test_vimba.py
@@ -0,0 +1,52 @@
+import pytest
+
+from pymba import Vimba, VimbaException
+
+
+def test_version():
+    version = Vimba().version.split('.')
+    assert int(version[0]) >= 1
+    assert int(version[1]) >= 7
+    assert int(version[2]) >= 0
+
+
+def test_startup_shutdown():
+    with pytest.raises(VimbaException) as e:
+        Vimba().system().feature_names()
+    assert e.value.error_code == VimbaException.ERR_STARTUP_NOT_CALLED
+
+    # manual
+    Vimba().startup()
+    Vimba().system().feature_names()
+    Vimba().shutdown()
+
+    # context manager
+    with Vimba() as vmb:
+        vmb.system().feature_names()
+
+
+@pytest.fixture
+def vimba() -> Vimba:
+    with Vimba() as vimba:
+        # for ethernet camera discovery
+        if vimba.system().GeVTLIsPresent:
+            vimba.system().run_feature_command("GeVDiscoveryAllOnce")
+        yield vimba
+
+
+def test_interface_camera_ids(vimba: Vimba):
+    # test id funcs return a list of strings (not bytes)
+    for func in (vimba.interface_ids, vimba.camera_ids):
+        ids = func()
+        assert isinstance(ids, list)
+        assert ids
+        for x in ids:
+            assert isinstance(x, str)
+
+
+def test_interface(vimba: Vimba):
+    interface = vimba.interface(vimba.interface_ids()[0])
+
+
+def test_camera(vimba: Vimba):
+    camera = vimba.camera(vimba.camera_ids()[0])