use unittest in test.py
complete