def test_recurrence_table(self, catalogue, config): '''Basic recurrence table test''' target_data = np.genfromtxt('data/recurrence_table_test_1.csv', delimiter = ',') self.assert(np.allclose(target_data, recurrence_table( self.catalogue['magnitude'], self.config['magnitude_interval'], self.catalogue['year'])))
def test_b_max_likelihood(self, catalogue, config, completeness): '''Test the basic maximum likelihood function''' # Cut the catalogue cut_catalogue = self._cut_catalogue_to_completeness() rec_table = recurrence_table(cut_catalogue['magnitude'], self.config['magnitude_interval'], cut_catalogue['year']) bvalue, sigmab = b_max_likelihood(rec_table[:, 0], rec_table[:, 1], self.config['magnitude_interval'] np.min(cut_catalogue['magnitude'])) self.assertAlmostEqual(bvalue, 1.02823300686, places=5) self.assertAlmostEqual(sigmab, 0.03262690837, places=5)