fix jwt tests

Author: parkererickson-tg

tests/test_jwtAuth.py

@@ -3,7 +3,8 @@
 import json
 
 from pyTigerGraphUnitTest import make_connection
-from pyTigerGraph import TigerGraphConnection, pyTigerGraphException
+from pyTigerGraph import TigerGraphConnection
+from pyTigerGraph.pyTigerGraphException import TigerGraphException
 
 
 class TestJWTTokenAuth(unittest.TestCase):
@@ -43,7 +44,7 @@ def _requestJWTToken(self):
 
 
     def _test_jwtauth_3_9(self):
-        with self.assertRaises(pyTigerGraphException) as context:
+        with self.assertRaises(TigerGraphException) as context:
             TigerGraphConnection(
                 host=self.conn.host,
                 jwtToken="fake.JWT.Token"
@@ -79,7 +80,7 @@ def _test_jwtauth_4_1_success(self):
 
 
     def _test_jwtauth_4_1_fail(self):
-        with self.assertRaises(pyTigerGraphException) as context:
+        with self.assertRaises(TigerGraphException) as context:
             TigerGraphConnection(
                 host=self.conn.host,
                 jwtToken="invalid.JWT.Token"