changes

tedivm · Nov 21, 2024 · c1c0b4d · c1c0b4d
1 parent fa69117
commit c1c0b4d
Showing 1 changed file with 2 additions and 17 deletions.
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -144,21 +144,6 @@ def test_version():
     assert result.exit_code == 0
 
 
-from click.testing import CliRunner
-from paracelsus.cli import app
-
-from sqlalchemy import Column, Integer, String
-from sqlalchemy.ext.declarative import declarative_base
-
-Base = declarative_base()
-
-
-class Comments(Base):
-    __tablename__ = "comments"
-    id = Column(Integer, primary_key=True)
-    text = Column(String)
-
-
 def test_graph_with_inclusion_regex(package_path: Path):
     result = runner.invoke(
         app,
@@ -170,7 +155,7 @@ def test_graph_with_inclusion_regex(package_path: Path):
             "--python-dir",
             str(package_path),
             "--include-tables",
-            "comments",
+            "^com.*",
         ],
     )
     assert result.exit_code == 0
@@ -190,7 +175,7 @@ def test_graph_with_exclusion_regex(package_path: Path):
             "--python-dir",
             str(package_path),
             "--exclude-tables",
-            "^pos*",
+            "^pos*.",
         ],
     )
     assert result.exit_code == 0