[WIP] add more tests

fixes #4