Refactor all tests with pytest

[huytran187]

Many references, but here's a quick one https://realpython.com/pytest-python-testing/