diff --git a/test_leap_year.py b/test_leap_year.py new file mode 100644 index 0000000..d00499e --- /dev/null +++ b/test_leap_year.py @@ -0,0 +1,20 @@ +import pytest + +from leap_year.leap_year import is_leap_year + + +@pytest.mark.parametrize("year,flag", [ + (1896, True), + (1900, False), + (1904, True), + + (1996, True), + (2000, True), + (2004, True), + + (2096, True), + (2100, False), + (2104, True), +]) +def test_leap_years(year, flag): + assert flag == is_leap_year(year) diff --git a/tox.ini b/tox.ini index e0a9e3a..8ea0ad6 100644 --- a/tox.ini +++ b/tox.ini @@ -1,5 +1,10 @@ [tox] -envlist = flake8 +envlist = py35,flake8 + + +[testenv] +deps = pytest +commands = pytest test_leap_year.py [testenv:flake8]