test.py

def test_cleanup():

    cleaned_objects = []

    class A:
        def cleanup(self):
            cleaned_objects.append(self)

    objects = main.init([A, A, A])
    main.cleanup(objects)
    assert cleaned_objects == objects[::-1]