test: flatten: add more behaviors