Add initial test for class definitions

**This is a breaking change and will require a version bump to v0.2.0**