Add Token.__eq__ to ease unit testing.

pekkaklarck · pekkaklarck · commit c81a5b4698eb · 2020-01-11T22:09:17.000+02:00
diff --git a/src/robot/parsing/lexer/tokens.py b/src/robot/parsing/lexer/tokens.py
@@ -123,6 +123,18 @@ def __repr__(self):
         return 'Token(%s, %r, %s, %s)' % (self.type, self.value,
                                           self.lineno, self.col_offset)
 
+    def __eq__(self, other):
+        if not isinstance(other, Token):
+            return False
+        return (self.type == other.type and
+                self.value == other.value and
+                self.lineno == other.lineno and
+                self.col_offset == other.col_offset and
+                self.error == other.error)
+
+    def __ne__(self, other):
+        return not self == other
+
 
 class EOS(Token):
     __slots__ = []
diff --git a/utest/parsing/test_lexer.py b/utest/parsing/test_lexer.py
@@ -19,12 +19,7 @@ def assert_tokens(source, expected, get_tokens=get_tokens, data_only=False):
                  % (len(expected), expected, len(tokens), tokens),
                  values=False)
     for act, exp in zip(tokens, expected):
-        exp = Token(*exp)
-        assert_equal(act.type, exp.type)
-        assert_equal(act.value, exp.value, formatter=repr)
-        assert_equal(act.lineno, exp.lineno)
-        assert_equal(act.col_offset, exp.col_offset)
-        assert_equal(act.end_col_offset, exp.col_offset + len(exp.value))
+        assert_equal(act, Token(*exp), formatter=repr)
 
 
 class TestName(unittest.TestCase):
diff --git a/utest/parsing/test_splitter.py b/utest/parsing/test_splitter.py
@@ -26,12 +26,7 @@ def verify_split(string, *expected_statements, **config):
         for act, exp in zip(tokens, expected):
             if exp[0] == DATA:
                 expected_data[-1].append(exp)
-            exp = Token(*exp)
-            assert_equal(act.type, exp.type)
-            assert_equal(act.value, exp.value)
-            assert_equal(act.lineno, exp.lineno)
-            assert_equal(act.col_offset, exp.col_offset)
-            assert_equal(act.end_col_offset, exp.col_offset + len(exp.value))
+            assert_equal(act, Token(*exp), formatter=repr)
     if not config:
         verify_split(string, *expected_data, data_only=True)
 
