Skip tests if psycogp2-binary is not installed

VOLTTRON · Aug 7, 2020 · 7f16a2d · 7f16a2d
1 parent 010324f
commit 7f16a2d
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/volttrontesting/platform/dbutils/test_postegresql_timescaledb.py b/volttrontesting/platform/dbutils/test_postegresql_timescaledb.py
@@ -4,9 +4,15 @@
 
 from time import time
 
-from psycopg2.sql import SQL, Identifier
 import pytest
-import psycopg2
+try:
+    import psycopg2
+    from psycopg2.sql import SQL, Identifier
+except ImportError:
+    pytest.skip(
+        "Required imports for testing are not installed; thus, not running tests.",
+        allow_module_level=True,
+    )
 
 from volttron.platform.dbutils.postgresqlfuncts import PostgreSqlFuncts
 from volttrontesting.fixtures.docker_wrapper import create_container