1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
import pytest
from scripts.use_io_common_urlopen import use_io_common_urlopen
PATH = "t.py"
def test_inconsistent_usage(capsys) -> None:
content = "from urllib.request import urlopen"
result_msg = (
"t.py:1:0: Don't use urllib.request.urlopen, "
"use pandas.io.common.urlopen instead\n"
)
with pytest.raises(SystemExit, match=None):
use_io_common_urlopen(content, PATH)
expected_msg, _ = capsys.readouterr()
assert result_msg == expected_msg
def test_consistent_usage() -> None:
# should not raise
content = "from pandas.io.common import urlopen"
use_io_common_urlopen(content, PATH)
|
