Test code base clean up

- Cleaned up tests code base.