add type hinting to test codes