def test_parse_cmp_subdivision(self):
with open('test_data/fms_api/2015_event_list.json', 'r') as f:
events, districts = FMSAPIEventListParser(2015).parse(
json.loads(f.read()))
event = events[3]
self.assertEquals(event.key_name, "2015tes")
self.assertEquals(event.name, "Tesla Division")
self.assertEquals(event.short_name, "Tesla")
self.assertEquals(event.event_short, "tes")
self.assertEquals(event.official, True)
self.assertEquals(
event.start_date,
datetime.datetime(year=2015,
month=4,
day=22,
hour=0,
minute=0,
second=0))
self.assertEquals(
event.end_date,
datetime.datetime(year=2015,
month=4,
day=25,
hour=23,
minute=59,
second=59))
self.assertEquals(event.venue, "Edward Jones Dome")
self.assertEquals(event.city, "St. Louis")
self.assertEquals(event.state_prov, "MO")
self.assertEquals(event.country, "USA")
self.assertEquals(event.year, 2015)
self.assertEquals(event.event_type_enum, EventType.CMP_DIVISION)
self.assertEquals(event.district_key, None)
def test_parse_regional_event(self):
with open('test_data/fms_api/2015_event_list.json', 'r') as f:
events, districts = FMSAPIEventListParser(2015).parse(
json.loads(f.read()))
event = events[0]
self.assertEquals(event.key_name, "2015nyny")
self.assertEquals(event.name, "New York City Regional")
self.assertEquals(event.short_name, "New York City")
self.assertEquals(event.event_short, "nyny")
self.assertEquals(event.official, True)
self.assertEquals(
event.start_date,
datetime.datetime(year=2015,
month=3,
day=12,
hour=0,
minute=0,
second=0))
self.assertEquals(
event.end_date,
datetime.datetime(year=2015,
month=3,
day=15,
hour=23,
minute=59,
second=59))
self.assertEquals(event.venue, "Jacob K. Javits Convention Center")
self.assertEquals(event.city, "New York")
self.assertEquals(event.state_prov, "NY")
self.assertEquals(event.country, "USA")
self.assertEquals(event.year, 2015)
self.assertEquals(event.event_type_enum, EventType.REGIONAL)
self.assertEquals(event.district_key, None)
def test_parse_offseason(self):
with open('test_data/fms_api/2015_event_list.json', 'r') as f:
events, districts = FMSAPIEventListParser(2015).parse(
json.loads(f.read()))
event = events[4]
self.assertEquals(event.key_name, "2015iri")
self.assertEquals(event.name, "Indiana Robotics Invitational")
self.assertEquals(event.short_name,
"Indiana Robotics Invitational")
self.assertEquals(event.event_short, "iri")
self.assertEquals(event.official, False)
self.assertEquals(
event.start_date,
datetime.datetime(year=2015,
month=7,
day=17,
hour=0,
minute=0,
second=0))
self.assertEquals(
event.end_date,
datetime.datetime(year=2015,
month=7,
day=18,
hour=23,
minute=59,
second=59))
self.assertEquals(event.venue, "Lawrence North HS")
self.assertEquals(event.city, "Indianapolis")
self.assertEquals(event.state_prov, "IN")
self.assertEquals(event.country, "USA")
self.assertEquals(event.year, 2015)
self.assertEquals(event.event_type_enum, EventType.OFFSEASON)
self.assertEquals(event.district_key, None)
def test_parse_district_cmp(self):
with open('test_data/fms_api/2015_event_list.json', 'r') as f:
events = FMSAPIEventListParser(2015).parse(json.loads(f.read()))
event = events[2]
self.assertEquals(event.key_name, "2015necmp")
self.assertEquals(
event.name,
"NE FIRST District Championship presented by United Technologies"
)
self.assertEquals(event.short_name, "NE FIRST")
self.assertEquals(event.event_short, "necmp")
self.assertEquals(event.official, True)
self.assertEquals(
event.start_date,
datetime.datetime(year=2015,
month=4,
day=8,
hour=0,
minute=0,
second=0))
self.assertEquals(
event.end_date,
datetime.datetime(year=2015,
month=4,
day=11,
hour=23,
minute=59,
second=59))
self.assertEquals(event.venue, "Sports and Recreation Center, WPI")
self.assertEquals(event.location, "Worcester, MA, USA")
self.assertEquals(event.year, 2015)
self.assertEquals(event.event_type_enum, EventType.DISTRICT_CMP)
self.assertEquals(event.event_district_enum,
DistrictType.NEW_ENGLAND)
def test_parse_district_event(self):
with open('test_data/fms_api/2015_event_list.json', 'r') as f:
events = FMSAPIEventListParser(2015).parse(json.loads(f.read()))
event = events[1]
self.assertEquals(event.key_name, "2015cthar")
self.assertEquals(event.name, "NE District - Hartford Event")
self.assertEquals(event.short_name, "Hartford")
self.assertEquals(event.event_short, "cthar")
self.assertEquals(event.official, True)
self.assertEquals(
event.start_date,
datetime.datetime(year=2015,
month=3,
day=27,
hour=0,
minute=0,
second=0))
self.assertEquals(
event.end_date,
datetime.datetime(year=2015,
month=3,
day=29,
hour=23,
minute=59,
second=59))
self.assertEquals(event.venue, "Hartford Public High School")
self.assertEquals(event.location, "Hartford, CT, USA")
self.assertEquals(event.year, 2015)
self.assertEquals(event.event_type_enum, EventType.DISTRICT)
self.assertEquals(event.event_district_enum,
DistrictType.NEW_ENGLAND)
def test_parse_2018_event(self):
with open('test_data/fms_api/2018_event_list.json', 'r') as f:
events, districts = FMSAPIEventListParser(2018).parse(json.loads(f.read()))
self.assertEqual(len(events), 178)
self.assertEqual(len(districts), 10)
event = events[18]
self.assertEquals(event.key_name, "2018casj")
self.assertEquals(event.name, "Silicon Valley Regional")
self.assertEquals(event.short_name, "Silicon Valley")
self.assertEquals(event.event_short, "casj")
self.assertEquals(event.official, True)
self.assertEquals(event.start_date, datetime.datetime(year=2018, month=3, day=28, hour=0, minute=0, second=0))
self.assertEquals(event.end_date, datetime.datetime(year=2018, month=3, day=31, hour=23, minute=59, second=59))
self.assertEquals(event.venue, "San Jose State University - The Event Center")
self.assertEquals(event.city, "San Jose")
self.assertEquals(event.state_prov, "CA")
self.assertEquals(event.country, "USA")
self.assertEquals(event.year, 2018)
self.assertEquals(event.event_type_enum, EventType.REGIONAL)
self.assertEquals(event.district_key, None)
self.assertEquals(event.website, "http://www.firstsv.org")
# New in 2018
self.assertEqual(event.webcast, [{"type": "twitch", "channel": "firstinspires9"}, {"type": "twitch", "channel": "firstinspires10"}])
def test_parse_2017_events_with_cmp_hacks(self):
hack_sitevar = Sitevar(id='cmp_registration_hacks')
hack_sitevar.contents = {
"event_name_override": [
{"event": "2017cmpmo", "name": "FIRST Championship Event", "short_name": "Championship"},
{"event": "2017cmptx", "name": "FIRST Championship Event", "short_name": "Championship"}],
"set_start_to_last_day": ["2017cmptx", "2017cmpmo"],
"divisions_to_skip": ["2017arc", "2017cars", "2017cur", "2017dal", "2017dar"],
}
hack_sitevar.put()
with open('test_data/fms_api/2017_event_list.json', 'r') as f:
events, districts = FMSAPIEventListParser(2017).parse(json.loads(f.read()))
self.assertEqual(len(events), 159)
self.assertEqual(len(districts), 10)
non_einstein_types = EventType.CMP_EVENT_TYPES
non_einstein_types.remove(EventType.CMP_FINALS)
for key in hack_sitevar.contents['divisions_to_skip']:
self.assertFalse(filter(lambda e: e.key_name == key, events))
einstein_stl = next(e for e in events if e.key_name == '2017cmpmo')
self.assertIsNotNone(einstein_stl)
self.assertEqual(einstein_stl.name, "FIRST Championship Event (St. Louis)")
self.assertEqual(einstein_stl.short_name, "Championship (St. Louis)")
self.assertEquals(einstein_stl.start_date, datetime.datetime(year=2017, month=4, day=29, hour=0, minute=0, second=0))
self.assertEquals(einstein_stl.end_date, datetime.datetime(year=2017, month=4, day=29, hour=23, minute=59, second=59))
einstein_hou = next(e for e in events if e.key_name == '2017cmptx')
self.assertIsNotNone(einstein_hou)
self.assertEqual(einstein_hou.name, "FIRST Championship Event (Houston)")
self.assertEqual(einstein_hou.short_name, "Championship (Houston)")
self.assertEquals(einstein_hou.start_date, datetime.datetime(year=2017, month=4, day=22, hour=0, minute=0, second=0))
self.assertEquals(einstein_hou.end_date, datetime.datetime(year=2017, month=4, day=22, hour=23, minute=59, second=59))
def test_parse_event_list(self):
with open('test_data/fms_api/2015_event_list.json', 'r') as f:
events = FMSAPIEventListParser(2015).parse(json.loads(f.read()))
self.assertTrue(isinstance(events, list))
# File has 5 events, but we ignore CMP divisions (only subdivisions), so only 4 are expected back
self.assertEquals(len(events), 4)
def test_parse_2017_official_offseason(self):
with open('test_data/fms_api/2017_event_list.json', 'r') as f:
events, districts = FMSAPIEventListParser(2017).parse(
json.loads(f.read()))
self.assertEqual(len(events), 165)
self.assertEqual(len(districts), 10)
event = next(e for e in events if e.key_name == "2017iri")
self.assertEquals(event.key_name, "2017iri")
self.assertEquals(event.name, "Indiana Robotics Invitational")
self.assertEquals(event.short_name,
"Indiana Robotics Invitational")
self.assertEquals(event.event_short, "iri")
self.assertEquals(event.official, True)
self.assertEquals(
event.start_date,
datetime.datetime(year=2017,
month=7,
day=14,
hour=0,
minute=0,
second=0))
self.assertEquals(
event.end_date,
datetime.datetime(year=2017,
month=7,
day=15,
hour=23,
minute=59,
second=59))
self.assertEquals(event.venue, "Lawrence North High School")
self.assertEquals(event.city, "Indianapolis")
self.assertEquals(event.state_prov, "IN")
self.assertEquals(event.country, "USA")
self.assertEquals(event.year, 2017)
self.assertEquals(event.event_type_enum, EventType.OFFSEASON)
self.assertEquals(event.district_key, None)
self.assertEquals(event.website,
"http://indianaroboticsinvitational.org/")
self.assertIsNone(event.webcast)
def test_parse_2017_event(self):
with open('test_data/fms_api/2017_event_list.json', 'r') as f:
events, districts = FMSAPIEventListParser(2017).parse(
json.loads(f.read()))
event = events[16]
self.assertEquals(event.key_name, "2017casj")
self.assertEquals(event.name, "Silicon Valley Regional")
self.assertEquals(event.short_name, "Silicon Valley")
self.assertEquals(event.event_short, "casj")
self.assertEquals(event.official, True)
self.assertEquals(
event.start_date,
datetime.datetime(year=2017,
month=3,
day=29,
hour=0,
minute=0,
second=0))
self.assertEquals(
event.end_date,
datetime.datetime(year=2017,
month=4,
day=1,
hour=23,
minute=59,
second=59))
self.assertEquals(event.venue,
"San Jose State University - The Event Center")
self.assertEquals(event.city, "San Jose")
self.assertEquals(event.state_prov, "CA")
self.assertEquals(event.country, "USA")
self.assertEquals(event.year, 2017)
self.assertEquals(event.event_type_enum, EventType.REGIONAL)
self.assertEquals(event.district_key, None)
# New in 2017
self.assertEquals(event.website, "http://www.firstsv.org")
def test_parse_division_parent(self):
with open('test_data/fms_api/2017_event_list.json', 'r') as f:
events, districts = FMSAPIEventListParser(2017).parse(
json.loads(f.read()))
self.assertEqual(len(events), 165)
self.assertEqual(len(districts), 10)
# Test division <-> parent associations
for event in events:
event_key = event.key.id()
if event_key == '2017micmp':
self.assertEqual(event.parent_event, None)
self.assertEqual(event.divisions, [
ndb.Key('Event', '2017micmp1'),
ndb.Key('Event', '2017micmp2'),
ndb.Key('Event', '2017micmp3'),
ndb.Key('Event', '2017micmp4'),
])
elif event_key in {
'2017micmp1', '2017micmp2', '2017micmp3', '2017micmp4'
}:
self.assertEqual(event.parent_event,
ndb.Key('Event', '2017micmp'))
self.assertEqual(event.divisions, [])
elif event_key == '2017cmptx':
self.assertEqual(event.parent_event, None)
self.assertEqual(event.divisions, [
ndb.Key('Event', '2017carv'),
ndb.Key('Event', '2017gal'),
ndb.Key('Event', '2017hop'),
ndb.Key('Event', '2017new'),
ndb.Key('Event', '2017roe'),
ndb.Key('Event', '2017tur'),
])
elif event_key in {
'2017carv', '2017gal', '2017hop', '2017new', '2017roe',
'2017tur'
}:
self.assertEqual(event.parent_event,
ndb.Key('Event', '2017cmptx'))
self.assertEqual(event.divisions, [])
elif event_key == '2017cmpmo':
self.assertEqual(event.parent_event, None)
self.assertEqual(event.divisions, [
ndb.Key('Event', '2017arc'),
ndb.Key('Event', '2017cars'),
ndb.Key('Event', '2017cur'),
ndb.Key('Event', '2017dal'),
ndb.Key('Event', '2017dar'),
ndb.Key('Event', '2017tes'),
])
elif event_key in {
'2017arc', '2017cars', '2017cur', '2017dal', '2017dar',
'2017tes'
}:
self.assertEqual(event.parent_event,
ndb.Key('Event', '2017cmpmo'))
self.assertEqual(event.divisions, [])
else:
self.assertEqual(event.parent_event, None)
self.assertEqual(event.divisions, [])
with open('test_data/fms_api/2018_event_list.json', 'r') as f:
events, districts = FMSAPIEventListParser(2018).parse(
json.loads(f.read()))
self.assertEqual(len(events), 178)
self.assertEqual(len(districts), 10)
# Test division <-> parent associations
for event in events:
event_key = event.key.id()
if event_key == '2018oncmp':
self.assertEqual(event.parent_event, None)
self.assertEqual(event.divisions, [
ndb.Key('Event', '2018oncmp1'),
ndb.Key('Event', '2018oncmp2'),
])
elif event_key in {'2018oncmp1', '2018oncmp2'}:
self.assertEqual(event.parent_event,
ndb.Key('Event', '2018oncmp'))
self.assertEqual(event.divisions, [])
elif event_key == '2018micmp':
self.assertEqual(event.parent_event, None)
self.assertEqual(event.divisions, [
ndb.Key('Event', '2018micmp1'),
ndb.Key('Event', '2018micmp2'),
ndb.Key('Event', '2018micmp3'),
ndb.Key('Event', '2018micmp4'),
])
elif event_key in {
'2018micmp1', '2018micmp2', '2018micmp3', '2018micmp4'
}:
self.assertEqual(event.parent_event,
ndb.Key('Event', '2018micmp'))
self.assertEqual(event.divisions, [])
elif event_key == '2018cmptx':
self.assertEqual(event.parent_event, None)
self.assertEqual(event.divisions, [
ndb.Key('Event', '2018carv'),
ndb.Key('Event', '2018gal'),
ndb.Key('Event', '2018hop'),
ndb.Key('Event', '2018new'),
ndb.Key('Event', '2018roe'),
ndb.Key('Event', '2018tur'),
])
elif event_key in {
'2018carv', '2018gal', '2018hop', '2018new', '2018roe',
'2018tur'
}:
self.assertEqual(event.parent_event,
ndb.Key('Event', '2018cmptx'))
self.assertEqual(event.divisions, [])
elif event_key == '2018cmpmi':
self.assertEqual(event.parent_event, None)
self.assertEqual(event.divisions, [
ndb.Key('Event', '2018arc'),
ndb.Key('Event', '2018cars'),
ndb.Key('Event', '2018cur'),
ndb.Key('Event', '2018dal'),
ndb.Key('Event', '2018dar'),
ndb.Key('Event', '2018tes'),
])
elif event_key in {
'2018arc', '2018cars', '2018cur', '2018dal', '2018dar',
'2018tes'
}:
self.assertEqual(event.parent_event,
ndb.Key('Event', '2018cmpmi'))
self.assertEqual(event.divisions, [])
else:
self.assertEqual(event.parent_event, None)
self.assertEqual(event.divisions, [])
def test_parse_2017_events_with_cmp_hacks(self):
hack_sitevar = Sitevar(id='cmp_registration_hacks')
hack_sitevar.contents = {
'should_store_divisions': False,
'einstein_name': 'FIRST Championship Event',
'einstein_short_name': 'Championship',
'should_change_einstein_dates': False
}
hack_sitevar.put()
with open('test_data/fms_api/2017_event_list.json', 'r') as f:
events, districts = FMSAPIEventListParser(2017).parse(
json.loads(f.read()))
self.assertEqual(len(events), 147)
self.assertEqual(len(districts), 10)
non_einstein_types = EventType.CMP_EVENT_TYPES
non_einstein_types.remove(EventType.CMP_FINALS)
self.assertFalse(
any(event.event_type_enum in non_einstein_types
for event in events))
einstein_stl = next(e for e in events if e.key_name == '2017cmpmo')
self.assertIsNotNone(einstein_stl)
self.assertEqual(einstein_stl.name,
"FIRST Championship Event (St. Louis)")
self.assertEqual(einstein_stl.short_name,
"Championship (St. Louis)")
self.assertEquals(
einstein_stl.start_date,
datetime.datetime(year=2017,
month=4,
day=26,
hour=0,
minute=0,
second=0))
self.assertEquals(
einstein_stl.end_date,
datetime.datetime(year=2017,
month=4,
day=29,
hour=23,
minute=59,
second=59))
einstein_hou = next(e for e in events if e.key_name == '2017cmptx')
self.assertIsNotNone(einstein_hou)
self.assertEqual(einstein_hou.name,
"FIRST Championship Event (Houston)")
self.assertEqual(einstein_hou.short_name, "Championship (Houston)")
self.assertEquals(
einstein_hou.start_date,
datetime.datetime(year=2017,
month=4,
day=19,
hour=0,
minute=0,
second=0))
self.assertEquals(
einstein_hou.end_date,
datetime.datetime(year=2017,
month=4,
day=22,
hour=23,
minute=59,
second=59))