Added a few tests for state vectors