Creates a demo test in pytest
# in a file called 'digits.py':
def last_digit(n1, n2):
'''
Define a function that takes in two non-negative integers a and b and
returns the last decimal digit of a^b
'''
return pow(n1, n2, 10)
# in a file called 'test_digits.py':
import pytest
from digits import last_digit
def test_last_digit():
assert last_digits(8, 2) == 4
assert last_digits(3, 2) == 9
# running a test:
# in the command line type python -m pytest test_digits.py
'''caveat:
Putting the tests in a subdirectory causes import issues (need to learn how to import from one directory above)
Not running pytest as a python module (python -m) causes name errors and import errors because it does not know where the python path is for some reason
'''