def test_time_for_url(time): # Create a fresh client, as fixtures don't work with @given LCClient = goes.XRSClient() time = time.start.strftime("%Y/%m/%d") almost_day = TimeDelta(1 * u.day - 1 * u.millisecond) tr = TimeRange(time, almost_day) url = LCClient._get_url_for_timerange(tr) times = LCClient._get_time_for_url(url) assert all([tr == t2 for t2 in times])
def LCClient(): return goes.XRSClient()
import pytest from sunpy.time.timerange import TimeRange, parse_time from sunpy.net.vso.attrs import Time, Instrument from sunpy.net.dataretriever.client import QueryResponse import sunpy.net.dataretriever.sources.goes as goes from sunpy.net.fido_factory import UnifiedResponse from sunpy.net import Fido from sunpy.net import attrs as a from hypothesis import given, example from sunpy.net.tests.strategies import goes_time LCClient = goes.XRSClient() @pytest.mark.parametrize( "timerange,url_start,url_end", [(TimeRange('1995/06/03', '1995/06/05'), 'http://umbra.nascom.nasa.gov/goes/fits/1995/go07950603.fits', 'http://umbra.nascom.nasa.gov/goes/fits/1995/go07950605.fits'), (TimeRange('2008/06/02', '2008/06/04'), 'http://umbra.nascom.nasa.gov/goes/fits/2008/go1020080602.fits', 'http://umbra.nascom.nasa.gov/goes/fits/2008/go1020080604.fits')]) def test_get_url_for_time_range(timerange, url_start, url_end): urls = LCClient._get_url_for_timerange(timerange) assert isinstance(urls, list) assert urls[0] == url_start assert urls[-1] == url_end