TST: Add test to verify the correct subclass

Thanks @ElBrogrammer & @Jorge-C