From 5853c6790678b012205a715cad38f500f7d3af5a Mon Sep 17 00:00:00 2001
From: Dave Halter <davidhalter88@gmail.com>
Date: Tue, 31 Dec 2019 18:49:18 +0100
Subject: [PATCH] Write tests for dict getitem

---
 test/test_api/test_interpreter.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/test/test_api/test_interpreter.py b/test/test_api/test_interpreter.py
index 3f5aeb49..e91523ef 100644
--- a/test/test_api/test_interpreter.py
+++ b/test/test_api/test_interpreter.py
@@ -606,3 +606,19 @@ def test_dict_completion(code, column, expected):
         expected = [e for e in expected if e is not Ellipsis]
 
     assert [c.complete for c in comps] == expected
+
+
+@pytest.mark.skipif(sys.version_info[0] == 2, reason="Ignore Python 2, because EOL")
+@pytest.mark.parametrize(
+    'code, types', [
+        ('dct[1]', ['int']),
+        ('dct["asdf"]', ['float']),
+        ('dct[r"asdf"]', ['float']),
+        ('dct["a"]', ['float', 'int']),
+    ]
+)
+def test_dict_getitem(code, types):
+    dct = {1: 2, "asdf": 1.0}
+
+    comps = jedi.Interpreter(code, [locals()]).infer()
+    assert [c.name for c in comps] == types