a little nicer wrt pep8

[sfa.git] / tests / testGid.py

diff --git a/tests/testGid.py b/tests/testGid.py
index 67c113b..36b91fe 100755 (executable)
--- a/tests/testGid.py
+++ b/tests/testGid.py
@@ -1,7 +1,6 @@
 import unittest
-import xmlrpclib
-from geni.trust.certificate import Keypair
-from geni.trust.gid import *
+from sfa.trust.certificate import Keypair
+from sfa.trust.gid import *
 
 class TestGid(unittest.TestCase):
    def setUp(self):
@@ -16,14 +15,14 @@ class TestGid(unittest.TestCase):
 
    def testSetGetUuid(self):
       gid = GID(subject="test")
-      u = create_uuid()
+      u = uuid.uuid4().int
 
       gid.set_uuid(u)
       self.assertEqual(gid.get_uuid(), u)
 
    def testEncodeDecode(self):
       gid = GID(subject="test")
-      u = str(uuid.uuid4().int)
+      u = uuid.uuid4().int
       hrn = "test.hrn"
 
       gid.set_uuid(u)
@@ -38,7 +37,7 @@ class TestGid(unittest.TestCase):
    def testSaveAndLoadString(self):
       gid = GID(subject="test")
 
-      u = str(uuid.uuid4().int)
+      u = uuid.uuid4().int
       hrn = "test.hrn"
 
       gid.set_uuid(u)