Add simple test using BayesNetwork class