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
