def test_load_from_array(self): # Tests the creation of a catalogue from an array and a key list cat = Catalogue() cat.load_from_array(['year', 'magnitude'], self.data_array) np.testing.assert_allclose(cat.data['magnitude'], self.data_array[:, 1]) np.testing.assert_allclose(cat.data['year'], self.data_array[:, 0].astype(int))
def test_load_to_array(self): """ Tests the creation of a catalogue from an array and a key list """ cat = Catalogue() cat.load_from_array(['year','magnitude'], self.data_array) data = cat.load_to_array(['year','magnitude']) self.assertTrue(np.allclose(data, self.data_array))
def test_load_to_array(self): """ Tests the creation of a catalogue from an array and a key list """ cat = Catalogue() cat.load_from_array(['year', 'magnitude'], self.data_array) data = cat.load_to_array(['year', 'magnitude']) self.assertTrue(np.allclose(data, self.data_array))
def test_load_from_array(self): """ Tests the creation of a catalogue from an array and a key list """ cat = Catalogue() cat.load_from_array(['year','magnitude'], self.data_array) self.assertTrue(np.allclose(cat.data['magnitude'],self.data_array[:,1])) self.assertTrue(np.allclose(cat.data['year'], self.data_array[:,0].astype(int)))
def test_catalogue_mt_filter(self): # Tests the catalogue magnitude-time filter cat = Catalogue() cat.load_from_array(['year', 'magnitude'], self.data_array) cat.data['eventID'] = np.arange(0, 7) cat.catalogue_mt_filter(self.mt_table) mag = np.array([7.0, 5.5, 5.01, 6.99]) yea = np.array([1920, 1970, 1960, 1960]) np.testing.assert_allclose(cat.data['magnitude'], mag) np.testing.assert_allclose(cat.data['year'], yea)
def test_load_from_array(self): """ Tests the creation of a catalogue from an array and a key list """ cat = Catalogue() cat.load_from_array(['year', 'magnitude'], self.data_array) self.assertTrue( np.allclose(cat.data['magnitude'], self.data_array[:, 1])) self.assertTrue( np.allclose(cat.data['year'], self.data_array[:, 0].astype(int)))
def test_catalogue_mt_filter(self): """ Tests the catalogue magnitude-time filter """ cat = Catalogue() cat.load_from_array(['year','magnitude'], self.data_array) cat.catalogue_mt_filter(self.mt_table) mag = np.array([7.0, 5.5, 5.01, 6.99]) yea = np.array([1920, 1970, 1960, 1960]) self.assertTrue(np.allclose(cat.data['magnitude'],mag)) self.assertTrue(np.allclose(cat.data['year'],yea))
def test_catalogue_mt_filter_no_flag(self): """ Tests the catalogue magnitude-time filter """ cat = Catalogue() cat.load_from_array(['year','magnitude'], self.data_array) cat.data['eventID'] = np.arange(0, len(cat.data['magnitude']), 1) cat.catalogue_mt_filter(self.mt_table) mag = np.array([7.0, 5.5, 5.01, 6.99]) yea = np.array([1920, 1970, 1960, 1960]) self.assertTrue(np.allclose(cat.data['magnitude'],mag)) self.assertTrue(np.allclose(cat.data['year'],yea))
def test_catalogue_mt_filter_with_flag(self): ''' Tests the catalogue magnitude-time filter when an input boolean vector is also defined ''' cat = Catalogue() cat.load_from_array(['year','magnitude'], self.data_array) cat.data['eventID'] = np.arange(0, len(cat.data['magnitude']), 1) flag = np.array([1, 1, 1, 1, 1, 0, 1], dtype=bool) cat.catalogue_mt_filter(self.mt_table, flag) mag = np.array([7.0, 5.5, 6.99]) yea = np.array([1920, 1970, 1960]) self.assertTrue(np.allclose(cat.data['magnitude'],mag)) self.assertTrue(np.allclose(cat.data['year'],yea))
def test_load_to_array(self): # Tests the creation of a catalogue from an array and a key list cat = Catalogue() cat.load_from_array(['year', 'magnitude'], self.data_array) data = cat.load_to_array(['year', 'magnitude']) np.testing.assert_allclose(data, self.data_array)