add state_matrix unit test
add base class tests