v1.01 unit tests

Updated test to support new data model.