How To Write A Unit Test Python