def test_get_remote_shake_map(self):
"""Check that we can retrieve both input and output from ftp at once.
"""
shake_event = '20110413170148'
shake_data = ShakeData(shake_event)
expected_input_file = os.path.join(shakemap_zip_dir(),
shake_event + '.inp.zip')
expected_output_file = os.path.join(shakemap_zip_dir(),
shake_event + '.out.zip')
if os.path.exists(expected_input_file):
os.remove(expected_input_file)
if os.path.exists(expected_output_file):
os.remove(expected_output_file)
input_file, output_file = shake_data.fetch_event()
message = ('Expected path for downloaded shakemap INP not received'
'\nExpected: %s\nGot: %s' %
(expected_output_file, output_file))
self.assertEqual(input_file, expected_input_file, message)
message = ('Expected path for downloaded shakemap OUT not received'
'\nExpected: %s\nGot: %s' %
(expected_output_file, output_file))
self.assertEqual(output_file, expected_output_file, message)
assert os.path.exists(expected_input_file)
assert os.path.exists(expected_output_file)
def test_get_remote_shake_map(self):
"""Check that we can retrieve both input and output from ftp at once"""
shake_event = '20110413170148'
shake_data = ShakeData(shake_event)
expected_input_file = os.path.join(
shakemap_zip_dir(),
shake_event + '.inp.zip')
expected_output_file = os.path.join(
shakemap_zip_dir(),
shake_event + '.out.zip')
if os.path.exists(expected_input_file):
os.remove(expected_input_file)
if os.path.exists(expected_output_file):
os.remove(expected_output_file)
input_file, output_file = shake_data.fetch_event()
message = ('Expected path for downloaded shakemap INP not received'
'\nExpected: %s\nGot: %s' %
(expected_output_file, output_file))
self.assertEqual(input_file, expected_input_file, message)
message = ('Expected path for downloaded shakemap OUT not received'
'\nExpected: %s\nGot: %s' %
(expected_output_file, output_file))
self.assertEqual(output_file, expected_output_file, message)
assert os.path.exists(expected_input_file)
assert os.path.exists(expected_output_file)
def setUp(self):
"""Copy our cached dataset from the fixture dir to the cache dir."""
output_file = '20120726022003.out.zip'
input_file = '20120726022003.inp.zip'
output_path = os.path.abspath(
os.path.join(os.path.dirname(__file__), '../fixtures',
output_file))
input_path = os.path.abspath(
os.path.join(os.path.dirname(__file__), '../fixtures', input_file))
shutil.copyfile(output_path,
os.path.join(shakemap_zip_dir(), output_file))
shutil.copyfile(input_path, os.path.join(shakemap_zip_dir(),
input_file))
def setUp(self):
"""Copy our cached dataset from the fixture dir to the cache dir."""
output_file = '20120726022003.out.zip'
input_file = '20120726022003.inp.zip'
output_path = os.path.abspath(os.path.join(
os.path.dirname(__file__),
'../fixtures',
output_file))
input_path = os.path.abspath(os.path.join(
os.path.dirname(__file__),
'../fixtures',
input_file))
shutil.copyfile(output_path,
os.path.join(shakemap_zip_dir(), output_file))
shutil.copyfile(input_path,
os.path.join(shakemap_zip_dir(), input_file))
def test_get_shake_map_output(self):
"""Check that we can retrieve a shakemap 'out' input file."""
event_id = '20110413170148'
shake_data = ShakeData(event_id)
shakemap_file = shake_data.fetch_output()
expected_file = os.path.join(shakemap_zip_dir(), event_id + '.out.zip')
message = 'Expected path for downloaded shakemap OUT not received'
self.assertEqual(shakemap_file, expected_file, message)
def test_get_shake_map_output(self):
"""Check that we can retrieve a shakemap 'out' input file"""
event_id = '20110413170148'
shake_data = ShakeData(event_id)
shakemap_file = shake_data.fetch_output()
expected_file = os.path.join(shakemap_zip_dir(),
event_id + '.out.zip')
message = 'Expected path for downloaded shakemap OUT not received'
self.assertEqual(shakemap_file, expected_file, message)
def test_get_shake_map_input(self):
"""Check that we can retrieve a shakemap 'inp' input file"""
shake_event = '20110413170148'
shake_data = ShakeData(shake_event)
shakemap_file = shake_data.fetch_input()
expected_file = os.path.join(shakemap_zip_dir(),
shake_event + '.inp.zip')
message = 'Expected path for downloaded shakemap INP not received'
self.assertEqual(shakemap_file, expected_file, message)
def test_get_latest_shake_map(self):
"""Check that we can retrieve the latest shake event"""
# Simply dont set the event id in the ctor to get the latest
shake_data = ShakeData()
input_file, output_file = shake_data.fetch_event()
event_id = shake_data.event_id
expected_input_file = os.path.join(shakemap_zip_dir(),
event_id + '.inp.zip')
expected_output_file = os.path.join(shakemap_zip_dir(),
event_id + '.out.zip')
message = ('Expected path for downloaded shakemap INP not received'
'\nExpected: %s\nGot: %s' %
(expected_output_file, output_file))
self.assertEqual(input_file, expected_input_file, message)
message = ('Expected path for downloaded shakemap OUT not received'
'\nExpected: %s\nGot: %s' %
(expected_output_file, output_file))
self.assertEqual(output_file, expected_output_file, message)
def test_get_cached_shake_map(self):
"""Check that we can retrieve both input and output from ftp at once"""
shake_event = '20120726022003'
expected_input_file = os.path.join(shakemap_zip_dir(),
shake_event + '.inp.zip')
expected_output_file = os.path.join(shakemap_zip_dir(),
shake_event + '.out.zip')
shake_data = ShakeData(shake_event)
input_file, output_file = shake_data.fetch_event()
message = ('Expected path for downloaded shakemap INP not received'
'\nExpected: %s\nGot: %s' %
(expected_output_file, output_file))
self.assertEqual(input_file, expected_input_file, message)
message = ('Expected path for downloaded shakemap OUT not received'
'\nExpected: %s\nGot: %s' %
(expected_output_file, output_file))
self.assertEqual(output_file, expected_output_file, message)
def test_get_shake_map_input(self):
"""Check that we can retrieve a shakemap 'inp' input file."""
shake_event = '20110413170148'
shake_data = ShakeData(shake_event)
shakemap_file = shake_data.fetch_input()
expected_file = os.path.join(shakemap_zip_dir(),
shake_event + '.inp.zip')
message = 'Expected path for downloaded shakemap INP not received'
self.assertEqual(shakemap_file, expected_file, message)
def test_get_latest_shake_map(self):
"""Check that we can retrieve the latest shake event."""
# Simply dont set the event id in the ctor to get the latest
shake_data = ShakeData()
input_file, output_file = shake_data.fetch_event()
event_id = shake_data.event_id
expected_input_file = os.path.join(shakemap_zip_dir(),
event_id + '.inp.zip')
expected_output_file = os.path.join(shakemap_zip_dir(),
event_id + '.out.zip')
message = ('Expected path for downloaded shakemap INP not received'
'\nExpected: %s\nGot: %s' %
(expected_output_file, output_file))
self.assertEqual(input_file, expected_input_file, message)
message = ('Expected path for downloaded shakemap OUT not received'
'\nExpected: %s\nGot: %s' %
(expected_output_file, output_file))
self.assertEqual(output_file, expected_output_file, message)
def test_get_cached_shake_map(self):
"""Check that we can retrieve both input and output from ftp at once.
"""
shake_event = '20120726022003'
expected_input_file = os.path.join(shakemap_zip_dir(),
shake_event + '.inp.zip')
expected_output_file = os.path.join(shakemap_zip_dir(),
shake_event + '.out.zip')
shake_data = ShakeData(shake_event)
input_file, output_file = shake_data.fetch_event()
message = ('Expected path for downloaded shakemap INP not received'
'\nExpected: %s\nGot: %s' %
(expected_output_file, output_file))
self.assertEqual(input_file, expected_input_file, message)
message = ('Expected path for downloaded shakemap OUT not received'
'\nExpected: %s\nGot: %s' %
(expected_output_file, output_file))
self.assertEqual(output_file, expected_output_file, message)
def test_shakemap_zip_dir(self):
"""Test we can get the shakemap zip dir."""
data_dir = shakemap_zip_dir()
expected_dir = '/tmp/inasafe/realtime/shakemaps-zipped'
assert os.path.exists(expected_dir)
self.assertEqual(data_dir, expected_dir)
def test_shakemap_zip_dir(self):
"""Test we can get the shakemap zip dir."""
data_dir = shakemap_zip_dir()
expected_dir = '/tmp/inasafe/realtime/shakemaps-zipped'
assert os.path.exists(expected_dir)
self.assertEqual(data_dir, expected_dir)
