tests/test_pag2edge.R

In pcalg: Methods for Graphical Models and Causal Inference

library(pcalg)
(doExtras <- pcalg:::doExtras())
if (doExtras) {
##################################################
## Mooij et al. (2020), Fig. 43(a), p. 97
##################################################

# Encode ADMG as adjacency matrix
p <- 8 # total number of variables
V <- c("Ca","Cb","Cc","X0","X1","X2","X3","X4") # 3 context variables, 5 system variables
# amat[i,j] = 0 iff no edge btw i,j
# amat[i,j] = 1 iff i *-o j
# amat[i,j] = 2 iff i *-> j
# amat[i,j] = 3 iff i *-- j
amat <- rbind(c(0,2,2,2,0,0,0,0),
              c(2,0,2,0,2,0,0,0),
              c(2,2,0,0,2,2,0,0),
              c(3,0,0,0,0,0,2,0),
              c(0,3,3,0,0,3,0,2),
              c(0,0,3,0,2,0,0,0),
              c(0,0,0,3,0,0,0,2),
              c(0,0,0,0,2,0,3,0))
rownames(amat)<-V
colnames(amat)<-V

# Make use of d-separation oracle as "independence test"
indepTest <- dsepAMTest
suffStat<-list(g=amat,verbose=FALSE)

# Derive PAG that represents the Markov equivalence class of the ADMG with the FCI algorithm
# (assuming no selection bias)
fci.pag <- fci(suffStat,indepTest,alpha = 0.5,labels = V,verbose=TRUE,selectionBias=FALSE)

# Read off causal features from the FCI PAG
cat('Identified absence (-1) and presence (+1) of direct causal relations from FCI PAG:\n')
fci.edge <- pag2edge(fci.pag@amat)
print(fci.edge)

true.edge <- rbind(c(-1, 0, 0, 0,-1,-1,-1,-1),
                   c( 0,-1, 0,-1, 0,-1,-1,-1),
                   c( 0, 0,-1,-1, 0, 0,-1,-1),
                   c( 0,-1,-1,-1,-1,-1, 0,-1),
                   c(-1,-1,-1,-1,-1,-1,-1,-1),
                   c(-1,-1, 0,-1, 0,-1,-1,-1),
                   c(-1,-1,-1, 0,-1,-1,-1, 0),
                   c(-1,-1,-1,-1,-1,-1,-1,-1))

rownames(true.edge)<-V
colnames(true.edge)<-V

stopifnot(identical(true.edge,fci.edge))
}