def setUp(self):
self.logger = dcs.Logger(0)
self.opti_opts = dcs.OptiOpts()
self.opti_opts.model_func = str
self.opti_opts.model_args = {'a': 1}
self.opti_opts.cost_func = str
self.opti_opts.cost_args = {'b': 2}
self.opti_opts.get_param_func = str
self.opti_opts.set_param_func = repr
self.opti_opts.output_folder = ''
self.opti_opts.output_results = ''
self.opti_opts.params = [1, 2]
def setUp(self):
self.logger = dcs.Logger(10)
time = np.arange(251)
sim_params = SimParams(time, magnitude=3.5, frequency=12, phase=180)
truth_time = np.arange(-10, 201)
truth_data = 5 * np.sin(2*np.pi*10*time/1000 + 90*np.pi/180)
self.opti_opts = dcs.OptiOpts()
self.opti_opts.model_func = sim_model
self.opti_opts.model_args = {'sim_params': sim_params}
self.opti_opts.cost_func = cost_wrapper
self.opti_opts.cost_args = {'results_time': time, 'truth_time': truth_time, 'truth_data': truth_data}
self.opti_opts.get_param_func = get_parameter
self.opti_opts.set_param_func = set_parameter
self.opti_opts.output_folder = ''
self.opti_opts.output_results = ''
self.opti_opts.params = []
# Parameters to estimate
self.opti_opts.params.append(dcs.OptiParam('magnitude', best=2.5, min_=-10, max_=10, typical=5, minstep=0.01))
self.opti_opts.params.append(dcs.OptiParam('frequency', best=20, min_=1, max_=1000, typical=60, minstep=0.01))
self.opti_opts.params.append(dcs.OptiParam('phase', best=180, min_=0, max_=360, typical=100, minstep=0.1))
def test_inequality2(self):
opti_opts = dcs.OptiOpts()
self.assertNotEqual(opti_opts, 2)
def test_inequality(self):
opti_opts1 = dcs.OptiOpts()
opti_opts2 = dcs.OptiOpts()
opti_opts2.grow_radius = 5.5
self.assertNotEqual(opti_opts1, opti_opts2)
def test_equality(self):
opti_opts1 = dcs.OptiOpts()
opti_opts2 = dcs.OptiOpts()
self.assertEqual(opti_opts1, opti_opts2)
def test_init(self):
opti_opts = dcs.OptiOpts()
self.assertTrue(isinstance(opti_opts, dcs.OptiOpts))
make_plots = True
time = np.arange(251)
# Parameters
sim_params = SimParams(time, magnitude=3.5, frequency=12, phase=180)
# Truth data
truth_time = np.arange(-10, 201)
truth_data = truth(truth_time)
truth = dcs.TruthPlotter(truth_time, truth_data)
# Logger
dcs.Logger().set_level(10)
# BPE Settings
opti_opts = dcs.OptiOpts()
opti_opts.model_func = sim_model
opti_opts.model_args = {'sim_params': sim_params}
opti_opts.cost_func = cost_wrapper
opti_opts.cost_args = {
'results_time': time,
'truth_time': truth_time,
'truth_data': truth_data
}
opti_opts.get_param_func = get_parameter
opti_opts.set_param_func = set_parameter
opti_opts.output_folder = os.path.join(dcs.get_output_dir(),
datetime.now().strftime('%Y-%m-%d'))
opti_opts.output_results = 'bpe_results.hdf5'
opti_opts.params = []