Converted existing sanity tests into unittest

Code is identical bar:
  * indentation 
  * assert is a unittest assert
  * removed prints