def test_init_complete_graph():
graph = PartialAncestralGraph(complete=True, variables=['A', 'B', 'C'])
assert graph.has_edge('A o-o B')
assert graph.has_edge('C o-o B')
assert graph.has_edge('A o-o C')
assert set({'A', 'C'}) == graph.get_neighbors('B')
assert set({'B', 'C'}) == graph.get_neighbors('A')
assert Edge('A o-o B') in graph.get_edges()
assert Edge('A o-o A') not in graph.get_edges()
assert Edge('B o-o B') not in graph.get_edges()
def test_2_multinom_RVs(df_2_multinomial_indep_RVs):
df = df_2_multinomial_indep_RVs(size=10000)
graph = Graph(variables=list(df.columns), complete=True)
skeleton_finder = PCSkeletonFinder(data=df, graph=graph)
cond_sets_satisfying_cond_indep = skeleton_finder.find()
assert graph.get_edges() == set({})
assert cond_sets_satisfying_cond_indep['x _||_ y'] == set({frozenset({})})