Improved test_subclass.py in ladim5