diff --git a/tests/test_new_mexico.py b/tests/test_new_mexico.py new file mode 100644 index 0000000..65fca42 --- /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"], "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"], 900) + self.assertEqual(game["num_tx_initial"], 670800)