changed unit test script to use new python class.