import nose.tools
from source.display import *

def test_display_start_of_game():
    display = Display()
    test = display.start
    assert test == "Welcome"

def test_display_game_over_draw():
    display = Display()
    test = display.draw
    assert test == "Draw"

def test_display_game_over_computer_wins():
    display = Display()
    test = display.computer
    assert test == "Computer Wins"

def test_display_game_over_human_wins():
    display = Display()
    test = display.human
    assert test == "You Win"

def test_display_current_board():
    pass

def test_display_computers_last_move():
    pass

def test_display_ask_for_human_input():
    display = Display()
    test = display.next_move
    assert test == "What is your next move?"

def test_display_human_made_illegal_move():
    pass

@@ -785,25 +785,115 @@
 illegal_move():%0A    
-pass
+display = Display()%0A    test = display.bad_move%0A    assert test == %22That is not a legal move.%22
 %0A

test display for illegal move message