blob: efb1a9dfcd76fa815933630d2e8229372bca886f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
from pathlib import Path
import subprocess
import sys
import alembic
from alembic.testing import eq_
from alembic.testing import TestBase
_home = Path(__file__).parent.parent
def run_command(file):
res = subprocess.run(
[
sys.executable,
str((_home / "tools" / "write_pyi.py").relative_to(_home)),
"--stdout",
"--file",
file,
],
stdout=subprocess.PIPE,
cwd=_home,
encoding="utf-8",
)
return res
class TestStubFiles(TestBase):
__requires__ = ("stubs_test",)
def test_op_pyi(self):
res = run_command("op")
generated = res.stdout
expected = Path(alembic.__file__).parent / "op.pyi"
eq_(generated, expected.read_text())
def test_context_pyi(self):
res = run_command("context")
generated = res.stdout
expected = Path(alembic.__file__).parent / "context.pyi"
eq_(generated, expected.read_text())
|
