def test_stone_color_feature(self): f = features.stone_color_feature(TEST_POSITION) self.assertEqual((9, 9, 3), f.shape) # plane 0 is B self.assertEqual(1, f[0, 1, 0]) self.assertEqual(0, f[0, 1, 1]) # plane 1 is W self.assertEqual(0, f[0, 8, 0]) self.assertEqual(1, f[0, 8, 1]) # plane 2 is empty self.assertEqual(1, f[0, 5, 2]) self.assertEqual(0, f[0, 5, 1])
def test_stone_color_feature(self): f = features.stone_color_feature(TEST_POSITION) self.assertEqual(f.shape, (9, 9, 3)) # plane 0 is B self.assertEqual(f[0, 1, 0], 1) self.assertEqual(f[0, 1, 1], 0) # plane 1 is W self.assertEqual(f[0, 8, 1], 1) self.assertEqual(f[0, 8, 0], 0) # plane 2 is empty self.assertEqual(f[0, 5, 2], 1) self.assertEqual(f[0, 5, 1], 0)