from unittest import TestCase

from go.helperboard import HelperBoard


class TestFlood(TestCase):
	def testCapture(self):
		b=[
			[0,1,0],
			[1,-1,1],
			[0,1,0]
		]
		h=HelperBoard(b)
		self.assertEqual(h.floodFill(-1,1,1,0), False)
		captured=set(h.getContinuousArea())
		self.assertEqual(captured, {(1,1)})