def testEventImport(self):
expected_result = ('category1', [
(20090308, 20090308, 'event1'),
(20090923, 20090925, 'event2'),
(20090111, 20090111, 'event3')
])
actual_result = eventutils.import_events("data/eventdata.txt")
self.assertEqual(actual_result, expected_result)
# Parse input data.
try:
data = utils.parse_data(input_file_path)
except IOError, err:
sys.exit(err)
# Import requested events.
# TODO(karol): this may have to be moved from here to the plotting
# file.
try:
events = { }
if import_catastrophic_events:
catastrophic_events = eventutils.import_events(
"../data/wydarzenia-katastrofy-polska.txt")
events[catastrophic_events[0]] = catastrophic_events[1]
if import_economical_events:
economical_events = eventutils.import_events(
"../data/wydarzenia-ekonomiczne-polska.txt")
events[economical_events[0]] = economical_events[1]
if import_other_events:
other_events = eventutils.import_events(
"../data/wydarzenia-inne-polska.txt")
events[other_events[0]] = other_events[1]
if import_political_events:
political_events = eventutils.import_events(
"../data/wydarzenia-polityczne-polska.txt")
events[political_events[0]] = political_events[1]
except IOError, err:
sys.exit(err)
