[tests] Import lit modules properly (via its package).

 - Lit itself ensures that its modules can be found, so this completely
   eliminates the need to modify PYTHONPATH.

 - This also makes it more obvious which parts of lit are being used in the
   custom format definition.