examples['rand32'] = dict(model=bdse_random(n=3, k=2), desc="") examples['indip1'] = dict(model=bdse_ex_one_command_indip(n=1, k=1), desc="") examples['indip3'] = dict(model=bdse_ex_one_command_indip(n=3, k=3), desc="") examples['indip_3_1'] = dict(model=bdse_ex_one_command_indip(n=3, k=1), desc="") return examples all_bdse_examples = bdse_examples() for_all_bdse_examples = fancy_test_decorator( lister=lambda: list(all_bdse_examples.keys()), arguments=lambda eid: (eid, all_bdse_examples[eid]['model']), attributes=lambda eid: dict(bdse_model=eid)) @for_all_bdse_examples def check_instantiation(mid, model): pass @for_all_bdse_examples def check_simulation(mid, model): # @UnusedVariable y0 = lambda: np.random.rand(model.get_y_shape()) u_dist = lambda: np.random.rand(model.get_u_shape()) simulator = BDSSimulator(model, y0, u_dist) for _ in simulator.get_simulation(10, 0.1): pass
''' Some utilities for testing. ''' from diffeo2d.diffeo_basic import diffeomorphism_from_function from diffeo2s.library import DiffeoLibrary from geometry.manifolds.tests.checks_generation import fancy_test_decorator for_all_diffeos = fancy_test_decorator( lister=lambda: list(DiffeoLibrary.diffeos.keys()), arguments=lambda fid: (fid, DiffeoLibrary.diffeos[fid]), attributes=lambda fid: dict(diffeo=fid)) def dd_lister(): return list(DiffeoLibrary.diffeos.keys()) def dd_get(id_diffeo): shape = (64, 64) func = DiffeoLibrary.diffeos[id_diffeo] D = diffeomorphism_from_function(shape, func) return (id_diffeo, D) for_all_dd = fancy_test_decorator(lister=dd_lister, arguments=dd_get, attributes=lambda ddid: dict(dd=ddid)) for_all_diffeo_pairs = fancy_test_decorator( lister=lambda: list(DiffeoLibrary.get_invertible_diffeos()), arguments=lambda pair: (pair[0].__name__, pair[0], pair[1].__name__, pair[1]), naming=lambda pair: '%s-%s' % (pair[0].__name__, pair[1].__name__))