def test_set_wind_data(self, all_winds): """ get_wind_data with default output format """ # check get_time_value() wm = Wind(timeseries=all_winds['wind'].get_wind_data(), coord_sys='r-theta', units='meter per second') gtime_val = wm.get_wind_data() x = gtime_val[:2] x['value'] = [(1, 10), (2, 20)] # default format is 'r-theta' wm.set_wind_data(x, 'meter per second') # only matches to 10^-14 assert np.allclose(wm.get_wind_data()['value'][:, 0], x['value'][:, 0], rtol=rtol, atol=atol) assert np.all(wm.get_wind_data()['time'] == x['time'])
def test_set_wind_data(self, all_winds): """ get_wind_data with default output format """ # check get_time_value() wm = Wind(timeseries=all_winds['wind'].get_wind_data(), format='r-theta', units='meter per second') gtime_val = wm.get_wind_data() x = gtime_val[:2] x['value'] = [(1, 10), (2, 20)] # default format is 'r-theta' wm.set_wind_data(x, 'meter per second') # only matches to 10^-14 assert np.allclose(wm.get_wind_data()['value'][:, 0], x['value'][:, 0], atol, rtol) assert np.all(wm.get_wind_data()['time'] == x['time'])
Test TypeError thrown if units are not given - so they are None """ # valid timeseries for testing dtv = np.zeros((4, ), dtype=datetime_value_2d).view(dtype=np.recarray) dtv.time = [datetime(2012, 11, 06, 20, 10 + i, 30,) for i in range(4)] dtv.value = (1, 0) # exception raised since no units given for timeseries during init with raises(TypeError): Wind(timeseries=dtv) # no units during set_timeseries with raises(TypeError): wind = Wind(timeseries=dtv, units='meter per second') wind.set_wind_data(dtv) # invalid units with raises(unit_conversion.InvalidUnitError): Wind(timeseries=dtv, units='met per second') def test_read_file_init(): """ initialize from a long wind file """ wm = Wind(filename=wind_file) print print '----------------------------------' print 'Units: ' + str(wm.units) assert True
11, 06, 20, 10 + i, 30, ) for i in range(4)] dtv.value = (1, 0) # exception raised since no units given for timeseries during init with raises(TypeError): Wind(timeseries=dtv) # no units during set_timeseries with raises(TypeError): wind = Wind(timeseries=dtv, units='meter per second') wind.set_wind_data(dtv) # invalid units with raises(unit_conversion.InvalidUnitError): Wind(timeseries=dtv, units='met per second') def test_read_file_init(): """ initialize from a long wind file """ wm = Wind(filename=wind_file) # have to test something: assert wm.units == 'knots'