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])