@@ -46,7 +46,7 @@ def test_rope_import_completion(config, workspace):
46
46
47
47
48
48
@pytest .mark .skipif (LooseVersion (jedi .__version__ ) < LooseVersion ('0.14.0' ),
49
- reason = 'This test fails with previous versions of jedi ' )
49
+ reason = 'This test fails with previous versions of Jedi ' )
50
50
def test_jedi_completion (config ):
51
51
# Over 'i' in os.path.isabs(...)
52
52
com_position = {'line' : 1 , 'character' : 15 }
@@ -118,3 +118,37 @@ def test_jedi_method_completion(config):
118
118
119
119
assert 'insertTextFormat' not in everyone_method
120
120
assert everyone_method ['insertText' ] == 'everyone'
121
+
122
+
123
+ @pytest .mark .skipif (LooseVersion ('0.15.0' ) <= LooseVersion (jedi .__version__ ) < LooseVersion ('0.16.0' ),
124
+ reason = 'This test fails with Jedi 0.15' )
125
+ def test_numpy_completions (config ):
126
+ doc_numpy = "import numpy as np; np."
127
+ com_position = {'line' : 0 , 'character' : len (doc_numpy )}
128
+ doc = Document (DOC_URI , doc_numpy )
129
+ items = pyls_jedi_completions (config , doc , com_position )
130
+
131
+ assert items
132
+ assert any (['array' in i ['label' ] for i in items ])
133
+
134
+
135
+ @pytest .mark .skipif (LooseVersion ('0.15.0' ) <= LooseVersion (jedi .__version__ ) < LooseVersion ('0.16.0' ),
136
+ reason = 'This test fails with Jedi 0.15' )
137
+ def test_pandas_completions (config ):
138
+ doc_pandas = "import pandas as pd; pd."
139
+ com_position = {'line' : 0 , 'character' : len (doc_pandas )}
140
+ doc = Document (DOC_URI , doc_pandas )
141
+ items = pyls_jedi_completions (config , doc , com_position )
142
+
143
+ assert items
144
+ assert any (['DataFrame' in i ['label' ] for i in items ])
145
+
146
+
147
+ def test_matplotlib_completions (config ):
148
+ doc_mpl = "import matplotlib.pyplot as plt; plt."
149
+ com_position = {'line' : 0 , 'character' : len (doc_mpl )}
150
+ doc = Document (DOC_URI , doc_mpl )
151
+ items = pyls_jedi_completions (config , doc , com_position )
152
+
153
+ assert items
154
+ assert any (['plot' in i ['label' ] for i in items ])
0 commit comments