move k nums test to its own module