Update tests to not require strict class names

[ZeuxWarrior]

Try to make the tests checking for specific class names to be more flexible. Maybe check what attributes are applied instead of the actual class name.

-M. Bugglin