|
| 1 | +import tempfile |
| 2 | + |
1 | 3 | import json
|
2 | 4 | import os
|
3 | 5 | import pprint
|
@@ -49,23 +51,28 @@ def _validate_serialization(original_model, model_deserialized):
|
49 | 51 |
|
50 | 52 | def test_save_model_to_disk():
|
51 | 53 | model = gp.generate_example_model(ExampleModel.COMBINATION, compute_model=False)
|
52 |
| - save_model(model, "temp/test_save_model_to_disk.gempy") |
| 54 | + with tempfile.NamedTemporaryFile(mode='w+', delete=True) as tmp: |
| 55 | + tmp_name = tmp.name + ".gempy" # Store the name to use it later |
| 56 | + save_model(model, tmp_name) |
53 | 57 |
|
54 |
| - # Load the model from disk |
55 |
| - loaded_model = load_model("temp/test_save_model_to_disk.gempy") |
| 58 | + # Load the model from disk |
| 59 | + loaded_model = load_model(tmp_name) |
56 | 60 | _validate_serialization(model, loaded_model)
|
57 | 61 |
|
58 | 62 | gp.compute_model(loaded_model)
|
59 | 63 | if True:
|
60 | 64 | import gempy_viewer as gpv
|
61 | 65 | gpv.plot_3d(loaded_model, image=True)
|
62 | 66 |
|
| 67 | + |
63 | 68 | # Test save after compute
|
64 |
| - save_model( |
65 |
| - model=model, |
66 |
| - path="temp/test_save_model_to_disk.gempy", |
67 |
| - validate_serialization=True |
68 |
| - ) |
| 69 | + with tempfile.NamedTemporaryFile(mode='w+', delete=True) as tmp: |
| 70 | + tmp_name = tmp.name + ".gempy" # Store the name to use it later |
| 71 | + save_model( |
| 72 | + model=model, |
| 73 | + path=tmp_name, |
| 74 | + validate_serialization=True |
| 75 | + ) |
69 | 76 |
|
70 | 77 |
|
71 | 78 | def test_interpolation_options():
|
|
0 commit comments