# import sys
# sys.path.insert(0, '/home/drobles/projects/mlnd-capstone-code/src/')
import unittest
from games import TicTacToe

class TestTicTacToe(unittest.TestCase):

    def setUp(self):
        self.game = TicTacToe()

    def test_cur_player_start(self):
        self.assertEqual(self.game.cur_player, 0)

    def test_cur_player_after_one_move(self):
        self.game.make_move(3)
        self.assertEqual(self.game.cur_player, 1)

    def test_is_not_over_at_start(self):
        self.assertFalse(self.game.is_over())

    def test_is_over_at_end_of_game(self):
        self.game.make_move(1)
        self.game.make_move(4)
        self.game.make_move(2)
        self.game.make_move(5)
        self.game.make_move(3)
        self.assertTrue(self.game.is_over())

    def test_cur_player_after_two_moves(self):
        self.game.make_move(3)
        self.game.make_move(7)
        self.assertEqual(self.game.cur_player, 0)

    def test_legal_moves_start(self):
        actual = self.game.legal_moves()
        expected = [1, 2, 3, 4, 5, 6, 7, 8, 9]
        self.assertItemsEqual(actual, expected)

    def test_legal_moves_after_one_move(self):
        self.game.make_move(1)
        actual = self.game.legal_moves()
        expected = [2, 3, 4, 5, 6, 7, 8, 9]
        self.assertItemsEqual(actual, expected)

    def test_legal_moves_after_two_moves(self):
        self.game.make_move(3)
        self.game.make_move(7)
        actual = self.game.legal_moves()
        expected = [1, 2, 4, 5, 6, 8, 9]
        self.assertItemsEqual(actual, expected)


if __name__ == 'main':
    unittest.main()

import unittest
from games import TicTacToe

class TestTicTacToe(unittest.TestCase):

    def setUp(self):
        self.game = TicTacToe()

    def test_cur_player_start(self):
        self.assertEqual(self.game.cur_player, 0)

    def test_cur_player_after_one_move(self):
        self.game.make_move(3)
        self.assertEqual(self.game.cur_player, 1)

    def test_is_not_over_at_start(self):
        self.assertFalse(self.game.is_over())

    def test_is_over_at_end_of_game(self):
        self.game.make_move(1)
        self.game.make_move(4)
        self.game.make_move(2)
        self.game.make_move(5)
        self.game.make_move(3)
        self.assertTrue(self.game.is_over())

    def test_cur_player_after_two_moves(self):
        self.game.make_move(3)
        self.game.make_move(7)
        self.assertEqual(self.game.cur_player, 0)

    def test_legal_moves_start(self):
        actual = self.game.legal_moves()
        expected = [1, 2, 3, 4, 5, 6, 7, 8, 9]
        self.assertItemsEqual(actual, expected)

    def test_legal_moves_after_one_move(self):
        self.game.make_move(1)
        actual = self.game.legal_moves()
        expected = [2, 3, 4, 5, 6, 7, 8, 9]
        self.assertItemsEqual(actual, expected)

    def test_legal_moves_after_two_moves(self):
        self.game.make_move(3)
        self.game.make_move(7)
        actual = self.game.legal_moves()
        expected = [1, 2, 4, 5, 6, 8, 9]
        self.assertItemsEqual(actual, expected)


if __name__ == 'main':
    unittest.main()

Remove commented imports