def test_average_difference_matrix_without_calc(self):
net = WTNetwork([[1, -1], [0, 1]], [0.5, 0])
states = list(net)
with_calc = net.average_sensitivity(states=deepcopy(states),
calc_trans=True)
without_calc = net.average_sensitivity(states=deepcopy(states),
calc_trans=False)
self.assertTrue(np.allclose(with_calc, without_calc, atol=1e-6))
def test_average_sensitivity(self):
net = WTNetwork([[1, -1], [0, 1]], [0.5, 0])
self.assertEqual(1.0, net.average_sensitivity())
def test_average_sensitivity_lengths(self):
net = WTNetwork([[1, -1], [0, 1]], [0.5, 0])
with self.assertRaises(ValueError):
net.average_sensitivity(states=[[0, 0], [0, 1]], weights=[0, 1, 2])
