tests, pyb: Add 'import pyb' when needed.