[test] add missing import math