test_evaluate -> test_inference

test/test_inference/test_literals.py

@@ -0,0 +1,46 @@
+import pytest
+from jedi.evaluate.context import TreeInstance
+
+
+def _eval_literal(Script, code, is_fstring=False):
+    def_, = Script(code).goto_definitions()
+    if is_fstring:
+        assert def_.name == 'str'
+        assert isinstance(def_._name._context, TreeInstance)
+        return ''
+    else:
+        return def_._name._context.get_safe_value()
+
+
+def test_f_strings(Script, environment):
+    """
+    f literals are not really supported in Jedi. They just get ignored and an
+    empty string is returned.
+    """
+    if environment.version_info < (3, 6):
+        pytest.skip()
+
+    assert _eval_literal(Script, 'f"asdf"', is_fstring=True) == ''
+    assert _eval_literal(Script, 'f"{asdf} "', is_fstring=True) == ''
+    assert _eval_literal(Script, 'F"{asdf} "', is_fstring=True) == ''
+    assert _eval_literal(Script, 'rF"{asdf} "', is_fstring=True) == ''
+
+
+def test_rb_strings(Script, environment):
+    assert _eval_literal(Script, 'br"asdf"') == b'asdf'
+    obj = _eval_literal(Script, 'rb"asdf"')
+
+    # rb is not valid in Python 2. Due to error recovery we just get a
+    # string.
+    assert obj == b'asdf'
+
+
+def test_thousand_separators(Script, environment):
+    if environment.version_info < (3, 6):
+        pytest.skip()
+
+    assert _eval_literal(Script, '1_2_3') == 123
+    assert _eval_literal(Script, '123_456_789') == 123456789
+    assert _eval_literal(Script, '0x3_4') == 52
+    assert _eval_literal(Script, '0b1_0') == 2
+    assert _eval_literal(Script, '0o1_0') == 8