def test_sorting_B(self):
dummyData = [('A',1,'M'),('B',2,'G'),('C',3,'A')]
self.assertEqual([dummyData[p] for p in [1,0,2]],
dfl_processor.sort_position_data(dummyData,'B'))
fname_pos = data_path + "ObservedPositionalData/BL/" + fname
matchname = os.path.splitext(fname)[0]
# Runs the parser
print "Parsing match information"
mip = prs.MatchInformationParser()
mip.run(fname_match)
teams, match = mip.getTeamInformation()
print "Parsing event data"
mep = prs.MatchEventParser()
mep.run(fname_info)
play_time, subs = mep.getEventInformation()
print "Parsing position data"
mpp = prs.MatchPositionParser(match,teams)
mpp.run(fname_pos)
pos_data,ball_data,time_stamps = mpp.getPositionInformation()
# Runs the processor
section = '2nd'
kk = pos_data['home'][section]
kks = prc.sort_position_data(kk)
bb = ball_data[section!='1st']
ss = prc.stitch_position_data(kks,bb)
data_transformed = prc.run(pos_data,ball_data,match)
# Runs the writer
print "Step 3: Saving Information\n"
matchdir(out_path+matchname)
wrt.write_data_to_file(data_transformed,matchname)
def test_sorting_A(self):
dummyData = [('A',1,'STZ'),('B',2,'DMZ'),('C',3,'LM')]
self.assertEqual([dummyData[p] for p in [1,2,0]],
dfl_processor.sort_position_data(dummyData))
fname_info = data_path + "Events/BL/" + fname
fname_pos = data_path + "ObservedPositionalData/BL/" + fname
matchname = os.path.splitext(fname)[0]
# Runs the parser
print("Parsing match information")
mip = prs.MatchInformationParser()
mip.run(fname_match)
teams, match = mip.getTeamInformation()
print("Parsing event data")
mep = prs.MatchEventParser()
mep.run(fname_info)
play_time, subs = mep.getEventInformation()
print("Parsing position data")
mpp = prs.MatchPositionParser(match, teams)
mpp.run(fname_pos)
pos_data, ball_data, time_stamps = mpp.getPositionInformation()
# Runs the processor
section = '2nd'
kk = pos_data['home'][section]
kks = prc.sort_position_data(kk)
bb = ball_data[section != '1st']
ss = prc.stitch_position_data(kks, bb)
data_transformed = prc.run(pos_data, ball_data, match)
# Runs the writer
print("Step 3: Saving Information\n")
matchdir(out_path + matchname)
wrt.write_data_to_file(data_transformed, matchname)
