reworked the tests to make them cleaner

The tests now use simple lists for the types to make it easier to
distinguish between all elements.
The tests themselves are also shared now.