how to create unittests for python prompt toolkit?,unittest.main()