def setUp(self):
"""
Read a sample catalogue containing 2 events after instantiating
the CsvCatalogueParser object.
"""
filename = os.path.join(self.BASE_DATA_PATH, 'test_gcmt_catalogue.txt')
parser = ParseNDKtoGCMT(filename)
parser_cent = ParseNDKtoGCMT(filename)
self.cat = parser.read_file()
self.cat_cent = parser_cent.read_file(use_centroid='True')
def setUp(self):
"""
Read a sample catalogue containing 2 events after instantiating
the CsvCatalogueParser object.
"""
filename = os.path.join(self.BASE_DATA_PATH, 'test_gcmt_catalogue.txt')
parser = ParseNDKtoGCMT(filename)
parser_cent = ParseNDKtoGCMT(filename)
self.cat = parser.read_file()
self.cat_cent = parser_cent.read_file(use_centroid='True')
def test_locations(self):
""" test locations """
fname = os.path.join(self.BASE_DATA_PATH, 'test_gcmt_catalogue_01.txt')
prs = ParseNDKtoGCMT(fname)
cat = prs.read_file(use_centroid=False)
self.assertAlmostEqual(cat.data['longitude'][0], 111.72)
self.assertAlmostEqual(cat.data['latitude'][0], -10.80)
self.assertAlmostEqual(cat.data['longitude'][1], -113.12)
self.assertAlmostEqual(cat.data['latitude'][1], 28.74)
cat = prs.read_file(use_centroid=True)
self.assertAlmostEqual(cat.data['longitude'][0], 112.44)
self.assertAlmostEqual(cat.data['latitude'][0], -10.60)
self.assertAlmostEqual(cat.data['longitude'][1], -113.12)
self.assertAlmostEqual(cat.data['latitude'][1], 28.81)
def set_gcmt(self, filename, bffer=75.):
"""
:parameter cmt_cat:
Name of a file in the .ndk format
"""
print('setting gcmt')
parser = ParseNDKtoGCMT(filename)
cmt_cat = parser.read_file()
loc = cmt_cat.data['longitude']
lac = cmt_cat.data['latitude']
minlo, maxlo, minla, maxla, qual = self.csec.get_mm()
if qual == 0:
idxs = self.csec.get_grd_nodes_within_buffer(loc,
lac,
bffer,
minlo, maxlo,
minla, maxla)
if qual == 1:
idxs = self.csec.get_grd_nodes_within_buffer_idl(loc,
lac,
bffer,
minlo, maxlo,
minla, maxla)
if idxs is not None:
cmt_cat.select_catalogue_events(idxs)
self.gcmt = cmt_cat
def setUp(self):
# Parse catalogue
fname = os.path.join(self.BASE_DATA_PATH, "test_gcmt_catalogue_01.txt")
# Parse GCMT file
prs = ParseNDKtoGCMT(fname)
self.cat = prs.read_file()
def test_without_specifying_years(self):
"""
Tests that when the catalogue is parsed without specifying the start
and end year that the start and end year come from the minimum and
maximum in the catalogue
"""
filename = os.path.join(self.BASE_DATA_PATH, 'test_gcmt_catalogue.txt')
parser = ParseNDKtoGCMT(filename)
self.cat = parser.read_file()
self.assertAlmostEqual(self.cat.start_year, np.min(self.cat.data['year']))
self.assertAlmostEqual(self.cat.end_year, np.max(self.cat.data['year']))
def test_without_specifying_years(self):
"""
Tests that when the catalogue is parsed without specifying the start
and end year that the start and end year come from the minimum and
maximum in the catalogue
"""
filename = os.path.join(self.BASE_DATA_PATH, 'test_gcmt_catalogue.txt')
parser = ParseNDKtoGCMT(filename)
self.cat = parser.read_file()
self.assertAlmostEqual(self.cat.start_year, np.min(self.cat.data['year']))
self.assertAlmostEqual(self.cat.end_year, np.max(self.cat.data['year']))
def get_dataframe(fname):
print(fname)
parser = ParseNDKtoGCMT(fname)
cat_gcmt = parser.read_file()
df = pd.DataFrame({k: cat_gcmt.data[k] for k in cat_gcmt.data.keys()})
return df
