Reformatted profile_workflow_classification_test and improved error m…

…essage
GEMScienceTools · Sep 30, 2024 · 8e434b6 · 8e434b6
1 parent 0c56b84
commit 8e434b6
Showing 1 changed file with 64 additions and 62 deletions.
diff --git a/openquake/sub/tests/profile_workflow_classification_test.py b/openquake/sub/tests/profile_workflow_classification_test.py
@@ -4,71 +4,73 @@
 from openquake.mbi.ccl import classify
 import tempfile
 import pathlib
-import os 
+import os
 import configparser
 import pandas as pd
 
 HERE = pathlib.Path(__file__).parent.resolve()
 
+
 def test_geojson_classification_workflow():
-	fname_geojson = HERE / 'data' / 'izu_slab2_css.geojson'
-	fname_slab = HERE / 'data' / 'izu_slab2_dep_02.24.18.grd'
-	classification_ini = HERE / 'data' / 'mariana_classification.ini'
-
-	## these go in a tmp folder...  
-	top_level = tempfile.mkdtemp()
-	profiles = tempfile.mkdtemp(dir=top_level)
-	sfc_out = tempfile.mkdtemp(dir=top_level)
-
-	# edit config to find the tmp file
-	config = configparser.ConfigParser()
-	config.read(classification_ini)
-	config.set('general', 'root_folder', top_level)
-
-	cat = HERE / 'data' / 'mariana_full_2202.pkl'
-	config.set('general', 'catalogue_filename', str(cat))
-
-	### This data comes from CRUST1.0, part of the Reference Earth Model
-	## (http://igppweb.ucsd.edu/~gabi/rem.html)
-	## Laske, G., Masters., G., Ma, Z. and Pasyanos, M., 
-	## Update on CRUST1.0 - A 1-degree Global Model of Earth's Crust,  
-	## Geophys. Res. Abstracts, 15, Abstract EGU2013-2658, 2013.
-	## Downloaded from https://igppweb.ucsd.edu/~gabi/crust1.html#download
-	crust = HERE / 'data' / 'depthtomoho.xyz'
-
-	config.set('crustal', 'crust_filename', str(crust))
-	with open(classification_ini, 'w') as configfile:
-		config.write(configfile)
-
-
-	max_sampl_dist = 25
-	upper_depth_int = 0
-	lower_depth_int = 50
-	lower_depth_slab = 300
-
-	slb = get_profiles_geojson(fname_geojson, fname_slab, spacing=10.)
-	slb.write_profiles(profiles)
-
-	create_2pt5_model(profiles, sfc_out, max_sampl_dist)
-
-	sfc_in = os.path.join(top_level, 'sfc_in')
-	os.makedirs(sfc_in)
-
-	sfc_sl = os.path.join(top_level, 'sfc_sl')
-	os.makedirs(sfc_sl)
-
-	build_complex_surface(sfc_out, max_sampl_dist, sfc_in,
-                              upper_depth_int, lower_depth_int)
-
-	build_complex_surface(sfc_out, max_sampl_dist, sfc_sl,
-                              lower_depth_int, lower_depth_slab)
-
-
-	classify.classify(classification_ini, True)
-
-	# Test the resulting classification is as expected
-	classified = pd.read_csv(os.path.join(top_level, 'classified_earthquakes.csv'))
-	test_set = pd.read_csv(HERE / 'data' /'classified_earthquakes_test.csv')
-
-	assert len(classified['tr']) == len(test_set['tr'])
-	assert all([a == b for a, b in zip(classified['tr'], test_set['tr'])])
+    fname_geojson = HERE / "data" / "izu_slab2_css.geojson"
+    fname_slab = HERE / "data" / "izu_slab2_dep_02.24.18.grd"
+    classification_ini = HERE / "data" / "mariana_classification.ini"
+
+    ## these go in a tmp folder...
+    top_level = tempfile.mkdtemp()
+    profiles = tempfile.mkdtemp(dir=top_level)
+    sfc_out = tempfile.mkdtemp(dir=top_level)
+
+    # edit config to find the tmp file
+    config = configparser.ConfigParser()
+    config.read(classification_ini)
+    config.set("general", "root_folder", top_level)
+
+    cat = HERE / "data" / "mariana_full_2202.pkl"
+    config.set("general", "catalogue_filename", str(cat))
+
+    ### This data comes from CRUST1.0, part of the Reference Earth Model
+    ## (http://igppweb.ucsd.edu/~gabi/rem.html)
+    ## Laske, G., Masters., G., Ma, Z. and Pasyanos, M.,
+    ## Update on CRUST1.0 - A 1-degree Global Model of Earth's Crust,
+    ## Geophys. Res. Abstracts, 15, Abstract EGU2013-2658, 2013.
+    ## Downloaded from https://igppweb.ucsd.edu/~gabi/crust1.html#download
+    crust = HERE / "data" / "depthtomoho.xyz"
+
+    config.set("crustal", "crust_filename", str(crust))
+    with open(classification_ini, "w") as configfile:
+        config.write(configfile)
+
+    max_sampl_dist = 25
+    upper_depth_int = 0
+    lower_depth_int = 50
+    lower_depth_slab = 300
+
+    slb = get_profiles_geojson(fname_geojson, fname_slab, spacing=10.0)
+    slb.write_profiles(profiles)
+
+    create_2pt5_model(profiles, sfc_out, max_sampl_dist)
+
+    sfc_in = os.path.join(top_level, "sfc_in")
+    os.makedirs(sfc_in)
+
+    sfc_sl = os.path.join(top_level, "sfc_sl")
+    os.makedirs(sfc_sl)
+
+    build_complex_surface(
+        sfc_out, max_sampl_dist, sfc_in, upper_depth_int, lower_depth_int
+    )
+
+    build_complex_surface(
+        sfc_out, max_sampl_dist, sfc_sl, lower_depth_int, lower_depth_slab
+    )
+
+    classify.classify(classification_ini, True)
+
+    # Test the resulting classification is as expected
+    classified = pd.read_csv(os.path.join(top_level, "classified_earthquakes.csv"))
+    test_set = pd.read_csv(HERE / "data" / "classified_earthquakes_test.csv")
+
+    assert len(classified["tr"]) == len(test_set["tr"])
+    for a, b in zip(classified["tr"], test_set["tr"]):
+        assert a == b