Changed tests to use unittest to allow for teardowns.