diff options
-rw-r--r-- | astroid/tests/unittest_python3.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/astroid/tests/unittest_python3.py b/astroid/tests/unittest_python3.py index e555bb48..2826a180 100644 --- a/astroid/tests/unittest_python3.py +++ b/astroid/tests/unittest_python3.py @@ -245,6 +245,20 @@ class Python3TC(unittest.TestCase): node = extract_node(code) self.assertEqual(node.as_string(), code) + @require_version('3.6') + def test_underscores_in_numeral_literal(self): + pairs = [ + ('10_1000', 101000), + ('10_000_000', 10000000), + ('0x_FF_FF', 65535), + ] + for value, expected in pairs: + node = extract_node(value) + inferred = next(node.infer()) + self.assertIsInstance(inferred, nodes.Const) + self.assertEqual(inferred.value, expected) + + if __name__ == '__main__': |