import unittest
from source.judge_pit import *
from source.game_table import *

class JudgeTestCase(unittest.TestCase):

    def setUp(self):
        self.human_1st_board = TableTop()
        self.computer_1st_board = TableTop()
        self.computer_1st_board.give_computer_the_first_move()
        self.human_1st_judge = Judge(self.human_1st_board)
        self.computer_1st_judge = Judge(self.computer_1st_board)
        self.mock_no_win_board = [1,10,1, 0,10,0, 1,0,10]

    def test_check_for_winner_returns_no_winner(self):
        self.human_1st_board.board = self.mock_no_win_board
        test_yields = self.human_1st_judge.check_for_winner()
        self.assertEqual(test_yields, False)

    def test_check_for_winner_returns_computer_p1_win(self):
        pass

    def test_check_for_winner_returns_computer_p2_win(self):
        pass

    def test_check_for_winner_returns_human_p1_win(self):
        pass

    def test_check_for_winner_returns_human_p2_win(self):
        pass

@@ -152,268 +152,131 @@
 elf.
-human_1st_board = TableTop()%0A        self.computer_1st_board = TableTop()%0A        self.computer_1st_board.give_computer_the_first_move()%0A        self.human_1st_judge = Judge(self.human_1st_board)%0A        self.computer_1st_judge = Judge(self.computer_1st_board)
+table_top = TableTop()%0A        self.judge = Judge(self.table_top)%0A        self.mock_no_win_board = %5B1,10,1, 0,10,0, 1,0,10%5D
 %0A   
@@ -282,34 +282,34 @@
       self.mock_
-no
+p1
 _win_board = %5B1,
@@ -306,33 +306,33 @@
 oard = %5B1,10,1, 
-0
+1
 ,10,0, 1,0,10%5D%0A%0A
@@ -331,16 +331,17 @@
 ,0,10%5D%0A%0A
+%0A
     def 
@@ -404,23 +404,17 @@
 elf.
-human_1st_board
+table_top
 .boa
@@ -472,18 +472,8 @@
 elf.
-human_1st_
 judg
@@ -600,36 +600,233 @@
 (self):%0A        
-pass
+self.table_top.board = self.mock_p1_win_board%0A        self.table_top.give_computer_the_first_move()%0A        test_yields = self.judge.check_for_winner()%0A        self.assertEqual(test_yields, 'computer')
 %0A%0A    def test_c

refactor judge tests