Change field names, add tests for coverage

tests/test_database.py

@@ -32,8 +32,8 @@ def test_powerball_draw_by_date(db_session) -> None:
     """Test to be sure we can find the first draw in the table by date."""
     with db_session as session:
         date = datetime(2010, 2, 3).date()
-        draw = db.PowerballDraw(draw_date=date, white_ball1=17, white_ball2=22, white_ball3=36, white_ball4=37,
+        draw = db.PowerballDraw(draw_date=date, main_ball1=17, main_ball2=22, main_ball3=36, main_ball4=37,
-             white_ball5=52, powerball=24, power_play=2)
+             main_ball5=52, powerball=24, power_play=2)
 
         record = session.query(db.PowerballDraw).filter_by(draw_date=date).first()
         assert record == draw
@@ -43,12 +43,25 @@ def test_powerball_draw_integrity(db_session) -> None:
     """Test that a proper error is received when a duplicate value insertion is attempted."""
     with db_session as session:
         date = datetime.strptime('02/03/2010','%m/%d/%Y').date()
+        draw = db.PowerballDraw(draw_date=date, main_ball1=17, main_ball2=22, main_ball3=36, main_ball4=37,
+                main_ball5=52, powerball=24, power_play=2)
 
         try:
-            draw = db.PowerballDraw(draw_date=date, white_ball1=17, white_ball2=22, white_ball3=36, white_ball4=37,
-                    white_ball5=52, powerball=24, power_play=2)
             session.add(draw)
             session.commit()
             assert False
         except IntegrityError:
             assert True
+        finally:
+            session.close()
+
+
+def test_powerball_repr_and_str(db_session) -> None:
+    """Test the __repr__ and __str__ methods of PowerballDraw objects."""
+    with db_session as session:
+        date = datetime(2010, 2, 3).date()
+        record = session.query(db.PowerballDraw).filter_by(draw_date=date).first()
+        assert str(record) == '2010-02-03, [17, 22, 36, 37, 52], 24, 2x'
+        assert repr(record) == 'PowerballDraw(self.draw_date=datetime.date(2010, 2, 3), self.main_ball1=17, ' \
+                'self.main_ball2=22, self.main_ball3=36, self.main_ball4=37, self.main_ball5=52, ' \
+                'self.powerball=24, self.power_play=2)'