From ac6338ba5be9a415721b719807ab98a24fad093f Mon Sep 17 00:00:00 2001 From: tdhood Date: Fri, 21 Apr 2023 14:44:03 -0700 Subject: [PATCH 1/2] added testing for new mexico --- tests/test_new_mexico.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 tests/test_new_mexico.py diff --git a/tests/test_new_mexico.py b/tests/test_new_mexico.py new file mode 100644 index 0000000..ab7820d --- /dev/null +++ b/tests/test_new_mexico.py @@ -0,0 +1,20 @@ +import unittest +import requests + +from lottery_data_scraper import new_mexico +from lottery_data_scraper import schemas + +class TestNewMexico(unittest.TestCase): + def test_parse_game_html(self): + # URL chosen arbitrarily + url = "https://www.nmlottery.com/games/scratchers" + html = new_mexico.fetch_html(url) + game = new_mexico.get_games(url)[0] + game = new_mexico.process_game(game) + self.assertEqual(game["name"], "$5,000 ORO") + self.assertEqual(game["price"], 2) + self.assertEqual(game["game_id"], "545") + self.assertEqual(game["prizes"][0]["prize"], "$5,000") + # Perhaps unfortunately in dollars. Cents would be better, eh? + self.assertEqual(game["prizes"][0]["value"], 5000) + self.assertEqual(game["num_tx_initial"], 498000) From e39445282f621d7e4304b20aba77b5eaf4eaea85 Mon Sep 17 00:00:00 2001 From: tdhood Date: Mon, 1 May 2023 16:28:13 -0700 Subject: [PATCH 2/2] updated new mexico test --- tests/test_new_mexico.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/test_new_mexico.py b/tests/test_new_mexico.py index ab7820d..65fca42 100644 --- a/tests/test_new_mexico.py +++ b/tests/test_new_mexico.py @@ -11,10 +11,10 @@ class TestNewMexico(unittest.TestCase): html = new_mexico.fetch_html(url) game = new_mexico.get_games(url)[0] game = new_mexico.process_game(game) - self.assertEqual(game["name"], "$5,000 ORO") - self.assertEqual(game["price"], 2) - self.assertEqual(game["game_id"], "545") - self.assertEqual(game["prizes"][0]["prize"], "$5,000") + self.assertEqual(game["name"], "Match 3 Tripler") + self.assertEqual(game["price"], 1) + self.assertEqual(game["game_id"], "521") + self.assertEqual(game["prizes"][0]["prize"], "$900") # Perhaps unfortunately in dollars. Cents would be better, eh? - self.assertEqual(game["prizes"][0]["value"], 5000) - self.assertEqual(game["num_tx_initial"], 498000) + self.assertEqual(game["prizes"][0]["value"], 900) + self.assertEqual(game["num_tx_initial"], 670800)