diff --git a/tests/conftest.py b/tests/conftest.py index 25ebbc1..3d984e3 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -23,6 +23,19 @@ class PowerballDraw(Base): power_play: Mapped[int] = mapped_column(Integer) +class MegaMillionsDraw(Base): + __tablename__ = 'mega_millions_draws' + + draw_date: Mapped[datetime.date] = mapped_column(Date, primary_key=True) + main_ball1: Mapped[int] = mapped_column(Integer) + main_ball2: Mapped[int] = mapped_column(Integer) + main_ball3: Mapped[int] = mapped_column(Integer) + main_ball4: Mapped[int] = mapped_column(Integer) + main_ball5: Mapped[int] = mapped_column(Integer) + mega_ball: Mapped[int] = mapped_column(Integer) + megaplier: Mapped[int] = mapped_column(Integer) + + @pytest.fixture(scope='session') def db_engine() -> Generator[Engine, Any, None]: # create a database connection engine