You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
21 lines
813 B
Python
21 lines
813 B
Python
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)
|