Commit 43619256 authored by Roman Yurchak's avatar Roman Yurchak

Check find_imports('import matplotlib.pyplot as plt')

parent 53de9fa5
......@@ -13,5 +13,6 @@ def test_find_imports():
import six
import numpy as np
from scipy import sparse
import matplotlib.pyplot as plt
"""))
assert set(res) == {'numpy', 'scipy', 'six'}
assert set(res) == {'numpy', 'scipy', 'six', 'matplotlib'}
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment