fix leap module/import for test runner

Also refactors the assignment implementation.