From dca8a4244e32b52e392ba0671d9eec486daa44c3 Mon Sep 17 00:00:00 2001 From: Chris Smith Date: Wed, 4 Mar 2026 12:57:35 -0500 Subject: [PATCH] Add MegaMillionsDraw to test config --- tests/conftest.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) 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