Make the unit test as simple as possible for now

* Rename functions to clearly state the feature being tested
* Add test to cope with TypeError on invalid argument