def test_main(self): cleanup.run_command = return_dpkg_query_and_uname_responses cleanup.output = result_collector cleanup.main() self.assertItemsEqual(result, EXPECTED)
#!/usr/bin/env python from ubuntu_old_kernel_cleanup import cleanup if __name__ == '__main__': cleanup.main()