Remove IME state assertions from Helper

To display additional error information, these assertions should be tests and not part of the transition.

Test: atest FlickerTests WMShellFlickerTests
Bug: 162925924
Change-Id: I94bd751a37487af49f9b12e55fce60305367200a