def test_negative_threshold(self):
xs = [0, 0, 0]
self.assertEqual(
[1, 0, 0], WTNetwork.negative_threshold([1, -1, 0], xs))
self.assertEqual([1, 0, 0], xs)
xs = [1, 1, 1]
self.assertEqual(
[1, 0, 0], WTNetwork.negative_threshold([1, -1, 0], xs))
self.assertEqual([1, 0, 0], xs)
def test_negative_threshold_scalar(self):
test = {
(1, 0): 1,
(0, 0): 0,
(-1, 0): 0,
(1, 1): 1,
(0, 1): 0,
(-1, 1): 0,
}
for x, s in test:
self.assertEqual(
test[(x, s)], WTNetwork.negative_threshold(x, s))