def test_source_model_to_hazardlib_converter(self): """ Tests the converter of a source model to an oq-hazardlib format """ # Load a full source model parser = nrmlSourceModelParser(MODEL_PATH) source_model = parser.read_file(2.0) oq_source_model = source_model.convert_to_oqhazardlib(TOM, 1.0, 10.0, 10.0, True) self.assertIsInstance(oq_source_model[0], AreaSource) self.assertIsInstance(oq_source_model[1], AreaSource) self.assertIsInstance(oq_source_model[2], PointSource) self.assertIsInstance(oq_source_model[3], SimpleFaultSource) self.assertIsInstance(oq_source_model[4], ComplexFaultSource) source_model.sources[3] = 'Rubbish!' with self.assertRaises(ValueError) as ver: oq_source_model = source_model.convert_to_oqhazardlib(TOM, 1.0, 10.0, 10.0, True) self.assertEqual(str(ver.exception), 'Source type not recognised!')
def test_source_model_to_hazardlib_converter(self): """ Tests the converter of a source model to an oq-hazardlib format """ # Load a full source model parser = nrmlSourceModelParser(MODEL_PATH) source_model = parser.read_file(2.0) oq_source_model = source_model.convert_to_oqhazardlib(TOM, 1.0, 10.0, 10.0, True) self.assertIsInstance(oq_source_model[0], AreaSource) self.assertIsInstance(oq_source_model[1], AreaSource) self.assertIsInstance(oq_source_model[2], PointSource) self.assertIsInstance(oq_source_model[3], SimpleFaultSource) self.assertIsInstance(oq_source_model[4], ComplexFaultSource) source_model.sources[3] = 'Rubbish!' with self.assertRaises(ValueError) as ver: oq_source_model = source_model.convert_to_oqhazardlib(TOM, 1.0, 10.0, 10.0, True) self.assertEqual(str(ver.exception), 'Source type not recognised!')
def test_nrml_writer(self): ''' Tests the source model writer ''' # Load a full source model parser = nrmlSourceModelParser(MODEL_PATH) source_model = parser.read_file(2.0) # Write to file source_model.serialise_to_nrml(TEST_PATH, True) # Load file back parser = nrmlSourceModelParser(TEST_PATH) source_model_test = parser.read_file(2.0) orig_sources = sorted(source_model.sources, key=operator.attrgetter('name')) test_sources = sorted(source_model_test.sources, key=operator.attrgetter('name')) for orig_source, test_source in zip(orig_sources, test_sources): self.assertEqual(orig_source.name, test_source.name) self.assertEqual(orig_source.mag_scale_rel.__class__.__name__, test_source.mag_scale_rel.__class__.__name__) # Remove the test file os.system('rm ' + TEST_PATH)
def test_nrml_writer(self): ''' Tests the source model writer ''' # Load a full source model parser = nrmlSourceModelParser(MODEL_PATH) source_model = parser.read_file(2.0) # Write to file source_model.serialise_to_nrml(TEST_PATH, True) # Load file back parser = nrmlSourceModelParser(TEST_PATH) source_model_test = parser.read_file(2.0) orig_sources = sorted(source_model.sources, key=operator.attrgetter('name')) test_sources = sorted(source_model_test.sources, key=operator.attrgetter('name')) for orig_source, test_source in zip(orig_sources, test_sources): self.assertEqual(orig_source.name, test_source.name) self.assertEqual(orig_source.mag_scale_rel.__class__.__name__, test_source.mag_scale_rel.__class__.__name__) # Remove the test file os.remove(TEST_PATH)
def test_execution_full(self): """ Tests the execution of the reader for the full model """ parser = nrmlSourceModelParser(self.nrml_file) model = parser.read_file("XXX") # Basic tests self.assertEqual(model.id, "XXX") self.assertEqual(model.name, "Some Source Model") self.assertEqual(len(model.sources), 5) # Area Source self._check_area_source(model.sources[0]) self._check_point_source(model.sources[2]) self._check_simple_fault_source(model.sources[3]) self._check_complex_fault_source(model.sources[4])
def test_execution_full(self): """ Tests the execution of the reader for the full model """ parser = nrmlSourceModelParser(self.nrml_file) model = parser.read_file("XXX") # Basic tests self.assertEqual(model.id, "XXX") self.assertEqual(model.name, "Some Source Model") self.assertEqual(len(model.sources), 5) # Area Source self._check_area_source(model.sources[0]) self._check_point_source(model.sources[2]) self._check_simple_fault_source(model.sources[3]) self._check_complex_fault_source(model.sources[4])