FileLocation: Correct exception class usage in comparison operators.

Use TypeError now (was: AttributeError).
behave · Nov 2, 2016 · 0e19c85 · 0e19c85
1 parent d518fc4
commit 0e19c85
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/behave/model_core.py b/behave/model_core.py
@@ -183,8 +183,8 @@ def __eq__(self, other):
         elif isinstance(other, six.string_types):
             return self.filename == other
         else:
-            raise AttributeError("Cannot compare FileLocation with %s:%s" % \
-                                 (type(other), other))
+            raise TypeError("Cannot compare FileLocation with %s:%s" % \
+                            (type(other), other))
 
     def __ne__(self, other):
         # return not self == other    # pylint: disable=unneeded-not
@@ -203,8 +203,8 @@ def __lt__(self, other):
         elif isinstance(other, six.string_types):
             return self.filename < other
         else:
-            raise AttributeError("Cannot compare FileLocation with %s:%s" % \
-                                 (type(other), other))
+            raise TypeError("Cannot compare FileLocation with %s:%s" % \
+                            (type(other), other))
 
     def __le__(self, other):
         # -- SEE ALSO: python2.7, functools.total_ordering