finished part way edits to the test function
(#297)