def test_unique_tag(): search = af.MockSearch() search.fit(model=af.Collection(), analysis=af.mock.mock.MockAnalysis()) identifier = search.paths.identifier search = af.MockSearch(unique_tag="dataset") search.fit( model=af.Collection(), analysis=af.mock.mock.MockAnalysis(), ) assert search.paths.identifier != identifier
def test_global_search(self, model_factor, model_factor_2): collection = g.FactorGraphModel(model_factor, model_factor_2) search = af.MockSearch() class Analysis(af.Analysis): def log_likelihood_function(self, instance): return collection.log_likelihood_function(instance) search.fit(collection.global_prior_model, Analysis())
def test_non_grid_searched_dimensions(self, mapper): grid_search = af.SearchGridSearch(number_of_steps=10, search=af.MockSearch()) grid_search.paths = af.DirectoryPaths(name="") mappers = list( grid_search.model_mappers( mapper, grid_priors=[mapper.component.one_tuple.one_tuple_0])) assert len(mappers) == 10 assert mappers[0].component.one_tuple.one_tuple_0.lower_limit == 0.0 assert mappers[0].component.one_tuple.one_tuple_0.upper_limit == 0.1 assert mappers[0].component.one_tuple.one_tuple_1.lower_limit == 0.0 assert mappers[0].component.one_tuple.one_tuple_1.upper_limit == 2.0 assert mappers[-1].component.one_tuple.one_tuple_0.lower_limit == 0.9 assert mappers[-1].component.one_tuple.one_tuple_0.upper_limit == 1.0 assert mappers[-1].component.one_tuple.one_tuple_1.lower_limit == 0.0 assert mappers[-1].component.one_tuple.one_tuple_1.upper_limit == 2.0
def test__move_pickle_files(self): search = af.MockSearch() search.paths = af.DirectoryPaths(name="pickles", path_prefix=path.join( "non_linear", "abstract_search")) pickle_paths = [ path.join(conf.instance.output_path, "non_linear", "abstract_search", "pickles") ] arr = np.ones((3, 3)) with open(path.join(pickle_paths[0], "test.pickle"), "wb") as f: pickle.dump(arr, f) pickle_paths = [ path.join( conf.instance.output_path, "non_linear", "abstract_search", "pickles", "test.pickle", ) ] search.paths._move_pickle_files(pickle_files=pickle_paths) with open(path.join(pickle_paths[0]), "rb") as f: arr_load = pickle.load(f) assert (arr == arr_load).all() if path.exists(test_path): shutil.rmtree(test_path)
def test_with_arguments(self): search = af.MockSearch() search.paths = af.DirectoryPaths(name="name") self.assert_paths_as_expected(search.paths)
def test_combination_argument(self): search = af.MockSearch("other", ) search.paths = af.DirectoryPaths(name="name") self.assert_paths_as_expected(search.paths)
def test_positional(self): search = af.MockSearch("name") paths = search.paths assert paths.name == "name"
def make_grid_search(mapper): search = af.SearchGridSearch( number_of_steps=10, search=af.MockSearch() ) search.paths = af.DirectoryPaths(name="") return search