Modify test to check engine was created.

Author: toddbirchard

tests/conftest.py

@@ -1,5 +1,7 @@
 """PyTest mocked fixtures."""
+
 import pytest
+
 from clients import Database
 from config import settings
 
@@ -8,7 +10,7 @@
 def db() -> Database:
     """Return a valid database object."""
     return Database(
-        uri = settings.SQLALCHEMY_DATABASE_URI,
-        table = settings.SQLALCHEMY_DATABASE_TABLE,
-        args = settings.SQLALCHEMY_CONNECT_ARGS,
+        uri=settings.SQLALCHEMY_DATABASE_URI,
+        table=settings.SQLALCHEMY_DATABASE_TABLE,
+        args=settings.SQLALCHEMY_CONNECT_ARGS,
     )

tests/test_table_fetch.py

@@ -1,11 +1,11 @@
 """Basic tests for validating app functionality."""
 
-from pandas import DataFrame
+from sqlalchemy.engine import Engine
 
 from clients.database import Database
 
 
 def test_fetch_sql_data(db: Database):
     """Test fetching data from a table."""
-    data = db.get_table_data()
-    assert isinstance(data, DataFrame)
+    db_engine = db.engine
+    assert isinstance(db_engine, Engine)