def check(self, test_name, tree_gen_func, **more_args): kws = self.some_kws_of_run.copy() kws.update(directed_params) if more_args: kws.update(more_args) paths = run( tree_gen_func, calculate_graph=False, print_summary=False, # result_pkl_path_prefix=result_pickle_prefix, **kws) trees = pkl.load(open(paths['result'])) trees = filter(lambda t: t.number_of_edges() > 0, trees) # remove empty trees assert_true(len(trees) > 0) for t in trees: assert_true(len(t.edges()) > 0) return trees, nx.read_gpickle(paths['meta_graph'])
def check(self, test_name, tree_gen_func, **more_args): kws = self.some_kws_of_run.copy() if more_args: kws.update(more_args) kws['root_sampling_method'] = 'random' paths = run(tree_gen_func, calculate_graph=False, print_summary=False, **kws) trees = pkl.load(open(paths['result'])) trees = filter(lambda t: t.number_of_edges() > 0, trees) # remove empty trees assert_true(len(trees) > 0) for t in trees: assert_true(len(t.edges()) > 0) return trees, nx.read_gpickle(paths['meta_graph'])