don't hide errors in test setup

TargetHolding · Jun 28, 2016 · 61cbedd · 61cbedd
1 parent 02ad8ec
commit 61cbedd
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/python/pyspark_cassandra/tests.py b/python/pyspark_cassandra/tests.py
@@ -671,6 +671,9 @@ def test_93(self):
         # unittest.TextTestRunner().run(suite)
     finally:
         # stop the spark context and cassandra session
-        CassandraTestCase.sc.stop()
-        CassandraTestCase.session.shutdown()
+        # stop the spark context and cassandra session
+        if hasattr(CassandraTestCase, 'sc'):
+            CassandraTestCase.sc.stop()
+        if hasattr(CassandraTestCase, 'session'):
+            CassandraTestCase.session.shutdown()