from $\_\_$future$\_\_$ import annotations

from four$\_$in$\_$a$\_$row import $*$

from a$0$ import $*$

# TODO add tests for each method and function as indicated in the assignment

# Note: we have scaffolded some code below for you to add your tests into.

# Make sure each test has a unique name and that each test starts

# with test$\_$

# The tests below are organized into classes to help keep related tests

# grouped together. In PyCharm you can choose to run all tests in a single

# class by using the run button beside the class name $($just like how you

# can choose to run a single test$).$ Alternatively, you can run all tests

# in the file by right$-$clicking the file name and choosing to run tests.

class TestHelpers:

$"""$

These are provided tests related to Task $1,$ which are meant to remind you

of the structure of a pytest for later tasks. For Task $1,$ you are asked

to write doctests instead.

While not required, you are welcome to add other pytests here as you

develop your code.

$"""$

def test$\_$within$\_$grid$\_$in$\_$grid$($self$)$:

$"""$Test that $(0,0)$ is inside a $4-$by$-4$ grid."""

assert within$\_$grid$((0,0),4)$

def test$\_$within$\_$grid$\_$outside$\_$grid$($self$)$:

$"""$Test that $(4,4)$ is outside a $4-$by$-4$ grid."""

assert not within$\_$grid$((4,4),4)$

def test$\_$all$\_$within$\_$grid$\_$all$\_$in$\_$grid$($self$)$:

$"""$Test when the four coordinates are all within a $4-$by$-4$ grid."""

assert all$\_$within$\_$grid$([(0,0),(1,1),(2,2),(3,3)],4)$

def test$\_$reflect$\_$vertically$\_$above$($self$)$:

$"""$Test reflecting vertically for a coordinate above the middle."""

assert reflect$\_$vertically$((0,1),5)==(4,1)$

def test$\_$reflect$\_$vertically$\_$middle$($self$)$:

$"""$Test reflecting vertically for a coordinate on the middle row."""

assert reflect$\_$vertically$((2,1),5)==(2,1)$

def test$\_$reflect$\_$points$($self$)$:

$"""$Test reflecting a very short line"""

assert reflect$\_$points$([(0,1),(1,2)],5)==[(4,1),(3,2)]$

class TestLine:

$"""$

TODO Task $2$: add tests for the Line methods and related functions

You must write two tests for each of:

$-$ is$\_$row, is$\_$column, and is$\_$diagonal

$-$ Line.drop, Line.is$\_$full, and Line.has$\_$fiar

$"""$

pass

class TestGrid:

$"""$

TODO Task $3\mathrm{.}1$: add tests for the Grid methods and related functions

You must write two tests for each of:

$-$ Grid.drop, Grid.is$\_$full

$-$ create$\_$rows$\_$and$\_$columns

TODO Task $3\mathrm{.}2$: add tests for the Grid methods and related functions

You must write two tests for each of:

$-$ Grid.has$\_$fiar

$-$ create$\_$mapping

$"""$

pass

class TestFourInARow:

$"""$

TODO TASK $4$:

$-$ run check$\_$coverage.py to get the code coverage report.

$-$ Using the code coverage report, identify which branches of the code

are not currently being tested.

$-$ add tests below in order to achieve $100\%$ code coverage when you run

check$\_$coverage.py$.$ Your tests should follow a similar structure

to the test$\_$x$\_$wins test defined below.

$"""$

def test$\_$x$\_$wins$($self$)->$ None:

$"""$

Provided test demonstrating how you can test FourInARow.play using

a StringIO object to "script" the input.

See both the handout and the Task $4$ section of the supplemental slides

for a detailed explanation of this example.

$"""$

fiar $=$ play$\_$game$($GAME$\_$SCRIPT$\_$X$\_$WINS$)$

assert fiar.result $==$ WIN

# TODO Add your tests for Task $4$ here. Make sure each test has a unique name

# and that each test starts with test$\_$

if $\_\_$name$\_\_==\text{'}\_\_$main$\_\_\text{'}$:

import pytest

pytest.main$([\text{'}$test$\_$a$0.$py$\text{'}])$