Stop pytest from picking up non-test classes

Stop pytest from picking up non-test classes #264