def test_get_random_key_and_seed(self): """ Test that get_random_key_and_seed functions as expected """ self.test_add_line_to_index() key, seed = get_random_key_and_seed(self.client, prefix=self.prefix) assert len(seed) == 2 assert self.prefix not in seed assert self.prefix in key
def test_get_completion(self): """ Test the get_completion method """ self.test_add_line_to_index() key, seed = get_random_key_and_seed(self.client, prefix=self.prefix) if STOP not in seed: assert get_completion(self.client, key) is not None else: assert get_completion(self.client, key) is None key = "test:i:ate" assert get_completion(self.client, key) in ["a", "one"] #ensure that exclude works as expected assert get_completion(self.client, key, exclude=["a",]) == "one" assert get_completion(self.client, key, exclude=["a","one"]) is None #ensure that relevant_terms works as well assert get_completion(self.client, key, relevant_terms=["one",]) == "one"
def test_get_completion(self): """ Test the get_completion method """ self.test_add_line_to_index() key, seed = get_random_key_and_seed(self.client, prefix=self.prefix) if STOP not in seed: assert get_completion(self.client, key) is not None else: assert get_completion(self.client, key) is None key = "test:i:ate" assert get_completion(self.client, key) in ["a", "one"] #ensure that exclude works as expected assert get_completion(self.client, key, exclude=[ "a", ]) == "one" assert get_completion(self.client, key, exclude=["a", "one"]) is None #ensure that relevant_terms works as well assert get_completion(self.client, key, relevant_terms=[ "one", ]) == "one"