Skip to content

Commit a9be7eb

Browse files
committed
Added a test for updating separated set
1 parent 1031806 commit a9be7eb

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

tests/TestSkeletonDiscovery.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
from unittest import TestCase
2+
import numpy as np
3+
from causallearn.search.ConstraintBased.PC import pc
4+
import networkx as nx
5+
from causallearn.utils.cit import chisq, fisherz, gsq, kci, mv_fisherz, d_separation
6+
7+
8+
class TestSkeletonDiscovery(TestCase):
9+
def test_sepset(self):
10+
truth_DAG_directed_edges = {(0, 2), (1, 2), (2, 3), (2, 4)}
11+
12+
true_dag_netx = nx.DiGraph()
13+
true_dag_netx.add_nodes_from(list(range(5)))
14+
true_dag_netx.add_edges_from(truth_DAG_directed_edges)
15+
16+
data = np.zeros((100, len(true_dag_netx.nodes))) # just a placeholder
17+
cg = pc(data, 0.05, d_separation, True, 0, -1, true_dag=true_dag_netx)
18+
assert cg.sepset[0, 2] is None

0 commit comments

Comments
 (0)