Add small_tests.py used to test small functions