modified test files for Interface handling and NotImplementedError

2 files changed, 9 insertions, 12 deletions

diff --git a/test/data/clientmodule_test.py b/test/data/clientmodule_test.py
index be4eaee..4efe45e 100644
--- a/test/data/clientmodule_test.py
+++ b/test/data/clientmodule_test.py
@@ -1,10 +1,10 @@
 """ docstring for file clientmodule.py """
-from data.suppliermodule_test import Interface as IFace, DoNothing
+from data.suppliermodule_test import Interface, DoNothing
 
-class Ancestor:
+class Ancestor(object):
     """ Ancestor method """
-    __implements__ = (IFace,)
-    
+    __implements__ = (Interface,)
+
     def __init__(self, value):
         local_variable = 0
         self.attr = 'this method shouldn\'t have a docstring'
@@ -17,11 +17,11 @@ class Ancestor:
     def set_value(self, value):
         self.__value = value
         return 'this method shouldn\'t have a docstring'
-    
+
 class Specialization(Ancestor):
     TYPE = 'final class'
     top = 'class'
-    
+
     def __init__(self, value, _id):
         Ancestor.__init__(self, value)
         self._id = _id
diff --git a/test/data/suppliermodule_test.py b/test/data/suppliermodule_test.py
index ddacb47..24dc9a0 100644
--- a/test/data/suppliermodule_test.py
+++ b/test/data/suppliermodule_test.py
@@ -1,13 +1,10 @@
 """ file suppliermodule.py """
 
-class NotImplemented(Exception):
-    pass
-
 class Interface:
     def get_value(self):
-        raise NotImplemented()
+        raise NotImplementedError
 
     def set_value(self, value):
-        raise NotImplemented()
+        raise NotImplementedError
 
-class DoNothing : pass
+class DoNothing: pass