Skip to content

Commit 065e01a

Browse files
committed
Add pandas and numpy import tests
1 parent dc1feb3 commit 065e01a

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

tests/checkers/test_numpy/test_import_numpy.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -9,17 +9,17 @@ class TestNumpyImport(pylint.testutils.CheckerTestCase):
99
CHECKER_CLASS = NumpyImportChecker
1010

1111
def test_correct_numpy_import(self):
12-
import_node = astroid.extract_node(
12+
numpy_import_node = astroid.extract_node(
1313
"""
1414
import numpy as np
1515
"""
1616
)
1717

1818
with self.assertNoMessages():
19-
self.checker.visit_import(import_node)
19+
self.checker.visit_import(numpy_import_node)
2020

2121
def test_incorrect_numpy_import(self):
22-
import_node = astroid.extract_node(
22+
numpy_import_node = astroid.extract_node(
2323
"""
2424
import numpy as npy
2525
"""
@@ -29,14 +29,14 @@ def test_incorrect_numpy_import(self):
2929
pylint.testutils.MessageTest(
3030
msg_id="numpy-import",
3131
confidence=HIGH,
32-
node=import_node,
32+
node=numpy_import_node,
3333
),
3434
ignore_position=True,
3535
):
36-
self.checker.visit_import(import_node)
36+
self.checker.visit_import(numpy_import_node)
3737

3838
def test_incorrect_numpy_import_from(self):
39-
importfrom_node = astroid.extract_node(
39+
numpy_importfrom_node = astroid.extract_node(
4040
"""
4141
from numpy import min
4242
"""
@@ -46,8 +46,8 @@ def test_incorrect_numpy_import_from(self):
4646
pylint.testutils.MessageTest(
4747
msg_id="numpy-importfrom",
4848
confidence=HIGH,
49-
node=importfrom_node,
49+
node=numpy_importfrom_node,
5050
),
5151
ignore_position=True,
5252
):
53-
self.checker.visit_importfrom(importfrom_node)
53+
self.checker.visit_importfrom(numpy_importfrom_node)

0 commit comments

Comments
 (0)