switched to use pytest for tests
fixed attribute name
added more tests