added testing for new mexico

main
Taylor Hood 2 years ago
parent b64fe3a307
commit ac6338ba5b

@ -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)
Loading…
Cancel
Save