Skip to content

Commit a8a185e

Browse files
TST: Fix tests with Environment class after exponential_backoff decorator was used
1 parent d0fbcd7 commit a8a185e

File tree

3 files changed

+6
-11
lines changed

3 files changed

+6
-11
lines changed

tests/fixtures/environment/environment_fixtures.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
from datetime import datetime, timedelta
22

33
import pytest
4+
45
from rocketpy import Environment, EnvironmentAnalysis
56

67

@@ -54,8 +55,8 @@ def env_analysis():
5455
EnvironmentAnalysis
5556
"""
5657
env_analysis = EnvironmentAnalysis(
57-
start_date=datetime.datetime(2019, 10, 23),
58-
end_date=datetime.datetime(2021, 10, 23),
58+
start_date=datetime(2019, 10, 23),
59+
end_date=datetime(2021, 10, 23),
5960
latitude=39.3897,
6061
longitude=-8.28896388889,
6162
start_hour=6,

tests/test_environment.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
import datetime
2-
import time
32
from unittest.mock import patch
43

54
import pytest
@@ -64,13 +63,8 @@ def test_wyoming_sounding_atmosphere(mock_show, example_plain_env):
6463
# "file" option, instead of receiving the URL as a string.
6564
URL = "http://weather.uwyo.edu/cgi-bin/sounding?region=samer&TYPE=TEXT%3ALIST&YEAR=2019&MONTH=02&FROM=0500&TO=0512&STNM=83779"
6665
# give it at least 5 times to try to download the file
67-
for i in range(5):
68-
try:
69-
example_plain_env.set_atmospheric_model(type="wyoming_sounding", file=URL)
70-
break
71-
except:
72-
time.sleep(1) # wait 1 second before trying again
73-
pass
66+
example_plain_env.set_atmospheric_model(type="wyoming_sounding", file=URL)
67+
7468
assert example_plain_env.all_info() == None
7569
assert abs(example_plain_env.pressure(0) - 93600.0) < 1e-8
7670
assert (

tests/unit/test_environment.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
1+
import json
12
import os
23

34
import numpy as np
45
import numpy.ma as ma
56
import pytest
67
import pytz
7-
import json
88

99
from rocketpy import Environment
1010

0 commit comments

Comments
 (0)