GEOS694_FinalProject/tests.py at main · swang331/GEOS694_FinalProject · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
# tests.py

import json
import tempfile
from pathlib import Path
from unittest.mock import patch

import profiles_update as pu


def create_valid_test_json(path):
    """
    Write a minimal valid G2S-style JSON file.

    This includes all required parameters expected by build_profile_dataframe():
      Z0, Z, T, U, V, R, P
    """
    test_data = {
        "metadata": {},
        "data": [
            {"parameter": "Z0", "values": [1.0]},
            {"parameter": "Z",  "values": [0.0, 1.0]},
            {"parameter": "T",  "values": [280.0, 275.0]},
            {"parameter": "U",  "values": [5.0, 6.0]},
            {"parameter": "V",  "values": [1.0, 2.0]},
            {"parameter": "R",  "values": [0.0012, 0.0011]},
            {"parameter": "P",  "values": [1000.0, 900.0]},
        ],
    }

    with open(path, "w") as f:
        json.dump(test_data, f)


def create_invalid_test_json(path):
    """
    Write an invalid JSON file missing pressure P.

    This is used to confirm the script raises an error when
    a required G2S parameter is missing.
    """
    test_data = {
        "metadata": {},
        "data": [
            {"parameter": "Z0", "values": [1.0]},
            {"parameter": "Z",  "values": [0.0, 1.0]},
            {"parameter": "T",  "values": [280.0, 275.0]},
            {"parameter": "U",  "values": [5.0, 6.0]},
            {"parameter": "V",  "values": [1.0, 2.0]},
            {"parameter": "R",  "values": [0.0012, 0.0011]},
        ],
    }

    with open(path, "w") as f:
        json.dump(test_data, f)


def test_main_creates_csv_for_valid_json():
    """
    Check that main() runs successfully for a valid JSON file
    and creates the expected CSV summary output.
    """
    with tempfile.TemporaryDirectory() as tmp:
        json_path = Path(tmp) / "sample.json"
        create_valid_test_json(json_path)

        # Patch runtime behavior so the test does not prompt for input
        # and does not try to open or save actual figures.
        with patch.object(pu, "download_g2s_json", return_value=json_path), \
             patch.object(pu, "RUN_AZIMUTH_PROMPT", False), \
             patch.object(pu, "RUN_PLOT_PROMPT", False), \
             patch.object(pu, "RUN_REFRACTION_FIGURE_PROMPT", False), \
             patch.object(pu, "SHOW_PLOTS", False), \
             patch.object(pu, "SAVE_PLOTS", False), \
             patch.object(pu, "SHOW_REFRACTION_FIGURE", False), \
             patch.object(pu, "SAVE_REFRACTION_FIGURE", False), \
             patch.object(pu, "plot_profile", return_value=None), \
             patch.object(pu, "plot_refraction_summary", return_value=None):

            pu.main()

        # The refactored script writes the CSV into a "summaries" folder
        output_csv = Path(tmp) / "summaries" / "sample_profiles.csv"
        assert output_csv.exists(), "Expected output CSV was not created"


def test_main_raises_error_for_missing_json_file():
    """
    Check that main() raises FileNotFoundError when the input JSON path
    does not exist.
    """
    fake_path = Path("/tmp/this_file_does_not_exist.json")

    with patch.object(pu, "download_g2s_json", return_value=fake_path):
        try:
            pu.main()
            assert False, "Expected FileNotFoundError"
        except FileNotFoundError:
            pass


def test_main_raises_error_for_missing_required_parameter():
    """
    Check that main() raises KeyError when the JSON file is missing
    a required G2S parameter.
    """
    with tempfile.TemporaryDirectory() as tmp:
        json_path = Path(tmp) / "bad.json"
        create_invalid_test_json(json_path)

        with patch.object(pu, "download_g2s_json", return_value=json_path), \
             patch.object(pu, "RUN_AZIMUTH_PROMPT", False), \
             patch.object(pu, "RUN_PLOT_PROMPT", False), \
             patch.object(pu, "RUN_REFRACTION_FIGURE_PROMPT", False), \
             patch.object(pu, "SHOW_PLOTS", False), \
             patch.object(pu, "SAVE_PLOTS", False), \
             patch.object(pu, "SHOW_REFRACTION_FIGURE", False), \
             patch.object(pu, "SAVE_REFRACTION_FIGURE", False), \
             patch.object(pu, "plot_profile", return_value=None), \
             patch.object(pu, "plot_refraction_summary", return_value=None):

            try:
                pu.main()
                assert False, "Expected KeyError"
            except KeyError:
                pass


if __name__ == "__main__":
    # Run the tests one by one.
    test_main_creates_csv_for_valid_json()
    test_main_raises_error_for_missing_json_file()
    test_main_raises_error_for_missing_required_parameter()

    print("All tests passed.")