class CatalogueTestCase(unittest.TestCase): def setUp(self): self.eqcat = EqCatalogue(IFACE) def _import_data(self, catfile, pattern): db_name = str(uuid.uuid1()) db = self.eqcat.create_db(catfile, pattern, db_name) try: self.eqcat.update_catalogue_db(db_name) finally: db.close() os.remove(db_name) def test_IASPEI(self): # no features with the default selections self._import_data(IASPEI, ImporterDialog.IASPEI_PATTERN) feature_count = self._filter_button( 'Events-2008:2008-5.0:8.0-mb,MS-IASPEI') self.assertEqual(feature_count, 0) # 58 features with magnitude in the range [1, 8] self.eqcat.dock.mag_range.setLowValue(1) self.eqcat.dock.mag_range.setHighValue(8) feature_count = self._filter_button( 'Events-2008:2008-1.0:8.0-mb,MS-IASPEI') self.assertEqual(feature_count,58) def test_ISF(self): self._import_data(ISF, ImporterDialog.ISF_PATTERN) feature_count = self._filter_button( 'Events-2010:2010-5.0:8.0-M,mb,mB,Mb,mb1,mb1mx,mbtmp,MD,ME,ML,MLv,' 'Ms,MS,Ms1,ms1mx,Ms7,Muk,Mw,MW-BJI,BKK,DJA,GCMT,GUC,IDC,IGQ,ISC,' 'ISCJB,JMA,MAN,MOS,NEIC,NIED,SJA,SZGRF') self.assertEqual(feature_count, 173) def test_load_countries(self): self.eqcat.load_countries() layerName = 'Population density' layer = QgsMapLayerRegistry.instance().mapLayersByName( layerName)[-1] self.assertEqual(layerName, str(layer.name())) def _filter_button(self, layername): btn = self.eqcat.dock.filterBtn QtTest.QTest.mouseClick(btn, QtCore.Qt.LeftButton) events = QgsMapLayerRegistry.instance().mapLayersByName( layername) assert events, 'Could not find layer named %s' % layername return events[0].featureCount()
def setUp(self): self.eqcat = EqCatalogue(IFACE)