def test_check_standard_model(self): """Test that standard model outputs nothing if there are no changes to the standard model """ model = sami2py.Model(tag='test', lon=256, year=1999, day=256, test=True) keys = model.check_standard_model() assert keys == list()
def test_model_repr(self): """Test that __repr__ returns a string of information.""" model = sami2py.Model(tag='test', lon=self.lon, year=self.year, day=self.day, test=True) repr_str = model.__repr__() assert type(repr_str) is str
def test_model_instantiation(self): """Test that model object is instantiated as a sami2py_model """ model = sami2py.Model(tag='test', lon=self.lon, year=self.year, day=self.day, test=True, outn=True) assert isinstance(model, sami2py.Model)
def test_to_netcdf(self): """Test that output file is correctly generated""" model = sami2py.Model(tag='test', lon=self.lon, year=self.year, day=self.day, test=True, outn=True) model.to_netcdf() path = 'sami2py_output.nc' savedat = xr.load_dataset(path) os.remove(path) assert model.data == savedat
def test_to_netcdf_w_path(self): """Test that output file is correctly generated""" model = sami2py.Model(tag='test', lon=self.lon, year=self.year, day=self.day, test=True, outn=True) path = 'custom_filename.nc' model.to_netcdf(path=path) savedat = xr.load_dataset('custom_filename.nc') os.remove(path) assert model.data == savedat
def test_check_standard_model(self): """Test that standard model outputs nothing if there are no changes to the standard model / changes to EUV in unformatted version """ model = sami2py.Model(tag='test', lon=self.lon, year=self.year, day=self.day, test=True) keys = model.check_standard_model() if self.day == 256: assert keys == list() else: assert 'EUV Multiplier' in keys
def test_model_plot(self): """Basic test that a reasonable plot has been created by testing the resulting axis limits """ import matplotlib matplotlib.use('Agg') import matplotlib.pyplot as plt model = sami2py.Model(tag='test', lon=256, year=1999, day=256, test=True) model.plot_lat_alt() fig = plt.gcf() xlims = fig.axes[0].get_xlim() ylims = fig.axes[0].get_ylim() assert xlims == (-36.3329, 19.37387) assert ylims == (84.98926, 1999.998) plt.close()
def test_model_plot(self): """Basic test that a reasonable plot has been created by testing the resulting axis limits """ import matplotlib model = sami2py.Model(tag='test', lon=self.lon, year=self.year, day=self.day, test=True) fig = model.plot_lat_alt() assert isinstance(fig, matplotlib.figure.Figure) tol = 1.e-4 xlims = fig.axes[0].get_xlim() ylims = fig.axes[0].get_ylim() assert abs(xlims[0] + 36.3329) < tol assert abs(xlims[1] - 19.37387) < tol assert abs(ylims[0] - 84.98926) < tol assert abs(ylims[1] - 1999.998) < tol matplotlib.pyplot.close(fig)
def test_model_input_exception(self): """File not found error should be produced if the file does not exist """ with pytest.raises(IOError): sami2py.Model(tag='none', lon=428, day=428, year=1969)
def test_model_instantiation_with_unformatted_files_and_mods(self): """Test that model object is instantiated as a sami2py_model """ model = sami2py.Model(tag='test', lon=256, year=1999, day=257, test=True) assert isinstance(model, sami2py.Model)
def test_model_instantiation(self): """Test that model object is instantiated as a sami2py_model """ model = sami2py.Model(tag='test', lon=256, year=1999, day=256, test=True) assert isinstance(model, sami2py.Model)