simplify tests with inheritance

I fucking love OOP sometimes