def test_correct_output(self, output_test_fixtures): params = output_test_fixtures.pop('params') output = output_test_fixtures.pop('output') output = output.magnitude * 1000 _to_si_units(params) _strip_units(params) assert_allclose(self.func(**params), output)
def test_correct_output(self, output_test_fixtures): params = output_test_fixtures.pop('params') outputs = output_test_fixtures.pop('output') _to_si_units(params) _strip_units(params) outputs = outputs.magnitude * 1000 from ...checks import check_correct_output_for_several_mus_and_sigmas check_correct_output_for_several_mus_and_sigmas( self.func, params, outputs)
def std_unitless_params(request, all_std_params): """ Returns set of standard params needed by requesting tested function. For using this fixture, the function test class needs to have a class attribute `func`, which is the tested function as a staticmethod. """ params = get_required_params(request.cls.func, all_std_params) _strip_units(params) return params
def test_correct_output(self, output_test_fixtures): params = output_test_fixtures.pop('params') output = output_test_fixtures.pop('output') _strip_units(params) output = output.magnitude / 1000 check_correct_output(self.func, params, output)
def test_V_0_larger_V_th_raise_exception(self, std_params): _strip_units(std_params) check_V_0_larger_V_th_raise_exception(self.func, std_params)
# only run code if users are sure they want to do it if '--force' in args.keys(): fixture_path = 'unit/data/' module = args['<module>'] config_path = 'unit/config/' + module + '/' param_files = os.listdir(config_path) regimes = [file.replace('.yaml', '') for file in param_files] regime_params = [load_val_unit_dict_from_yaml(config_path + file) for file in param_files] for dict in regime_params: _to_si_units(dict) _strip_units(dict) if module == 'firing_rates': create_and_save_fixtures(lmt.lif.delta._firing_rate, regime_params, regimes, fixture_path + 'lif_delta_firing_rate.h5') create_and_save_fixtures(lmt.lif.exp._firing_rate_taylor, regime_params, regimes, fixture_path + 'lif_exp_firing_rate_taylor.h5') create_and_save_fixtures(lmt.lif.exp._firing_rate_shift, regime_params, regimes, fixture_path + 'lif_exp_firing_rate_shift.h5') elif module == 'inputs': create_and_save_fixtures(lmt.lif._static._mean_input,