def setUpClass(self): np.random.seed(42) self.rep = 300 self.parallel = "seq" self.dbformat = "ram" self.timeout = 5 self.fig_name = 'test_output.png' sampler = spotpy.algorithms.mc(rosenbrock_setup(), sim_timeout=self.timeout) sampler.sample(self.rep) self.results = sampler.getdata() sampler = spotpy.algorithms.mc(griewank_setup(), sim_timeout=self.timeout) sampler.sample(self.rep) self.griewank_results = sampler.getdata() if sys.version_info >= (3, 6): # FAST is only fully operational under #Python 3 sampler = spotpy.algorithms.fast(rosenbrock_setup(), sim_timeout=self.timeout) sampler.sample(self.rep) self.sens_results = sampler.getdata() #Hymod resuts are empty with Python <3.6 sampler = spotpy.algorithms.dream(hymod_setup(GausianLike), sim_timeout=self.timeout) self.r_hat = sampler.sample(self.rep) self.hymod_results = sampler.getdata()
def test_plot_bestmodelrun(self): spotpy.analyser.plot_bestmodelrun(self.griewank_results, griewank_setup().evaluation(), fig_name=self.fig_name) # approximately 8855 KB is the size of an empty matplotlib.pyplot.plot, so # we expecting a plot with some content without testing the structure of the plot, just # the size self.assertGreaterEqual(os.path.getsize(self.fig_name), 8855) os.remove(self.fig_name)
def test_plot_parametertrace_algorithms(self): spotpy.analyser.plot_parametertrace_algorithms( [self.griewank_results], ["test_plot_parametertrace_algorithms"], spot_setup=griewank_setup(), fig_name=self.fig_name) # approximately 8855 KB is the size of an empty matplotlib.pyplot.plot, so # we expecting a plot with some content without testing the structure of the plot, just # the size self.assertGreaterEqual(os.path.getsize(self.fig_name), 8855) os.remove(self.fig_name)