def test_init(self): model = NoisyOrModel(['x1', 'x2', 'x3'], [2, 3, 2], [[0.6, 0.4], [0.2, 0.4, 0.7], [0.1, 0.4]]) np_test.assert_array_equal(model.variables, np.array(['x1', 'x2', 'x3'])) np_test.assert_array_equal(model.cardinality, np.array([2, 3, 2])) self.assertListEqual(model.inhibitor_probability, [[0.6, 0.4], [0.2, 0.4, 0.7], [0.1, 0.4]])
def setUp(self): self.model = NoisyOrModel(['x1', 'x2', 'x3'], [2, 3, 2], [[0.6, 0.4], [0.2, 0.4, 0.7], [0.1, 0.4]])