[PATCH 4/5] add tests