def basic_network(only_structure=False): """ Returns a basic valid network to test the APIs """ structure = { "node1": { "state1": 0.3, "state2": 0.7, }, "node2": { "state1": 0.7, "state2": 0.3, }, } if only_structure: return structure Net = BayesNet() # defining a valid network structure: Net.create_network(structure) return Net
def basic_network(only_structure=False): """ create basic network to test the APIs """ Net = BayesNet("continuous") network_struct = { "node1": { "states": ["dim1"], }, "node2": { "states": ["dim1", "dim2"], }, "node3": { "states": ["dim1", "dim2", "dim3"], }, "node4": { "states": ["dim1", "dim2", "dim3"], }, "node5": { "states": ["dim1"], }, } Net.create_network(network_struct) return Net
def setUp(self): self.Net = BayesNet("continuous") self.valid_structure = { "node1": { "states": ["dim1"], }, "node2": { "states": ["dim1", "dim2"], }, "node3": { "states": ["dim1", "dim2", "dim3"], }, "node4": { "states": ["dim1", "dim2", "dim3"], }, }
def setUp(self): self.Net = BayesNet() network_structure = { "node1": { "state1": 0.3, "state2": 0.7, }, "node2": { "state1": 0.7, "state2": 0.3, }, "node3": { "state1": 0.5, "state2": 0.5 }, } self.Net.create_network(network_structure)
def setUp(self): self.Net = BayesNet() # defining a valid network structure: self.valid_structure = { "node1": { "state1": 0.3, "state2": 0.7, }, "node2": { "state1": 0.7, "state2": 0.3, }, } self.invalid_structure = { "node1": { "state1": 0.3, "state2": "invalid probability", }, }