updated tests to have polymorphic tests

updated the tests to have polymorphic tests