Make the function tests Python3-import friendly

Change-Id: Ic73fc2a6f6712505eda71fa2e2e91ac680ced9a3