class CsvCatalogueParserTestCase(unittest.TestCase):
""" Unit tests for the csv Catalogue Parser Class"""
BASE_DATA_PATH = os.path.join(os.path.dirname(__file__), 'data')
def setUp(self):
self.gardner_knopoff_window = GardnerKnopoffWindow()
self.gruenthal_window = GruenthalWindow()
self.uhrhammer_window = UhrhammerWindow()
def test_gardner_knopoff_window(self):
"""
Test the Gardner and Knopoff Distance-Time window
"""
mag = np.array([5.0, 6.6])
sw_space, sw_time = self.gardner_knopoff_window.calc(mag)
self.assertAlmostEqual(sw_space[0], 39.99447, places=5)
self.assertAlmostEqual(sw_space[1], 63.10736, places=5)
self.assertAlmostEqual(sw_time[0], 143.71430 / 364.75, places=5)
self.assertAlmostEqual(sw_time[1], 891.45618 / 364.75, places=5)
def test_gruenthal_window(self):
"""
Test the Gruenthal Distance-Time window
"""
mag = np.array([5.0, 6.6])
sw_space, sw_time = self.gruenthal_window.calc(mag)
#self.assertAlmostEqual(sw_space[0], 39.99447, places=5)
#self.assertAlmostEqual(sw_time[0], 143.71430/364.75, places=5)
#self.assertAlmostEqual(sw_space[1], 63.10736, places=5)
#self.assertAlmostEqual(sw_time[1], 891.45618/364.75, places=5)
def test_uhrhammer_window(self):
mag = np.array([5.0, 6.6])
sw_space, sw_time = self.uhrhammer_window.calc(mag)
class CsvCatalogueParserTestCase(unittest.TestCase):
""" Unit tests for the csv Catalogue Parser Class"""
BASE_DATA_PATH = os.path.join(os.path.dirname(__file__), "data")
def setUp(self):
self.gardner_knopoff_window = GardnerKnopoffWindow()
self.gruenthal_window = GruenthalWindow()
self.uhrhammer_window = UhrhammerWindow()
def test_gardner_knopoff_window(self):
"""
Test the Gardner and Knopoff Distance-Time window
"""
mag = np.array([5.0, 6.6])
sw_space, sw_time = self.gardner_knopoff_window.calc(mag)
self.assertAlmostEqual(sw_space[0], 39.99447, places=5)
self.assertAlmostEqual(sw_space[1], 63.10736, places=5)
self.assertAlmostEqual(sw_time[0], 143.71430 / 364.75, places=5)
self.assertAlmostEqual(sw_time[1], 891.45618 / 364.75, places=5)
def test_gruenthal_window(self):
"""
Test the Gruenthal Distance-Time window
"""
mag = np.array([5.0, 6.6])
sw_space, sw_time = self.gruenthal_window.calc(mag)
# self.assertAlmostEqual(sw_space[0], 39.99447, places=5)
# self.assertAlmostEqual(sw_time[0], 143.71430/364.75, places=5)
# self.assertAlmostEqual(sw_space[1], 63.10736, places=5)
# self.assertAlmostEqual(sw_time[1], 891.45618/364.75, places=5)
def test_uhrhammer_window(self):
mag = np.array([5.0, 6.6])
sw_space, sw_time = self.uhrhammer_window.calc(mag)
plot_weekday_histogram(catalogue, color='#5fbdce', alpha=0.6, filename=output_base+"_weekday.png")
# hour_of_day plot
plot_hour_histogram(catalogue, color='#5fbdce', alpha=0.6, filename=output_base+"_hour.png")
# seismic_rate plot
# cumulative number of eq
if not declustered:
# decluster catalog
# plot decluster
# save decluster catalog
config = {
'decluster_method': 'gardner_knopoff', # afteran | gardner_knopoff
#'time_distance_window': GardnerKnopoffWindow(),
'time_distance_window': GruenthalWindow(),
#'time_distance_window': UhrhammerWindow(),
# afteran
'time_window': 60.,
# gardner_knopoff
'fs_time_prop': 1.0,
# write declustered
'filename': None,
#'filename': output_base+"_pp_decluster.csv",
# plot cluster
'plot':False,
'figname': output_base+"_pp_decluster.png",
}
declustered_catalogue = decluster_catalogue(catalogue, config)
plot_rate(catalogue, cumulative=True, color='#F0E797', new_figure=True, overlay=True, linewidth=1.5, label="Gardner-Knopoff/Gruenthal", alpha=0.8)
def setUp(self):
self.gardner_knopoff_window = GardnerKnopoffWindow()
self.gruenthal_window = GruenthalWindow()
self.uhrhammer_window = UhrhammerWindow()
def setUp(self):
self.gardner_knopoff_window = GardnerKnopoffWindow()
self.gruenthal_window = GruenthalWindow()
self.uhrhammer_window = UhrhammerWindow()
