def test_would_capture_feature(self):
f = features.would_capture_feature(TEST_POSITION2)
self.assertEqual(f.shape, (9, 9, features.would_capture_feature.planes))
# move at (1, 2) would capture 2 stones
self.assertEqual(f[1, 2, 1], 1)
# move at (0, 0) should not capture stones because it's B's move.
self.assertEqual(f[0, 0, 0], 0)
# move at (0, 7) would capture 3 stones
self.assertEqual(f[0, 7, 2], 1)
self.assertEqual(f[0, 7, 1], 0)
def test_would_capture_feature(self):
f = features.would_capture_feature(TEST_POSITION2)
self.assertEqual(f.shape, (9, 9, features.would_capture_feature.planes))
# move at (1, 2) would capture 2 stones
self.assertEqual(f[1, 2, 1], 1)
# move at (0, 0) should not capture stones because it's B's move.
self.assertEqual(f[0, 0, 0], 0)
# move at (0, 7) would capture 3 stones
self.assertEqual(f[0, 7, 2], 1)
self.assertEqual(f[0, 7, 1], 0)
def test_would_capture_feature(self):
f = features.would_capture_feature(TEST_POSITION2)
self.assertEqual((9, 9, features.would_capture_feature.planes),
f.shape)
# move at (1, 2) would capture 2 stones
self.assertEqualNPArray([1], f[1, 2])
# move at (0, 0) should not capture stones because it's B's move.
self.assertEqualNPArray([0], f[0, 0])
# move at (0, 7) would capture 3 stones
self.assertEqualNPArray([1], f[0, 7])