From 34ed8deaf02c752082b8fff69a5e058c3fb23ab2 Mon Sep 17 00:00:00 2001 From: Konsta Vesterinen Date: Tue, 14 Oct 2014 16:07:36 +0300 Subject: [PATCH] Skip encrypted type tests if cryptography not installed --- tests/types/test_encrypted.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/tests/types/test_encrypted.py b/tests/types/test_encrypted.py index e77a18b..002bc9b 100644 --- a/tests/types/test_encrypted.py +++ b/tests/types/test_encrypted.py @@ -1,11 +1,18 @@ import sqlalchemy as sa +from pytest import mark +cryptography = None +try: + import cryptography +except ImportError: + pass + from tests import TestCase from sqlalchemy_utils import EncryptedType -from sqlalchemy_utils.types.encrypted import AesEngine -from sqlalchemy_utils.types.encrypted import FernetEngine +from sqlalchemy_utils.types.encrypted import AesEngine, FernetEngine + +@mark.skipif('cryptography is None') class EncryptedTypeTestCase(TestCase): - def setup_method(self, method): # set some test values self.test_key = 'secretkey1234'