add test_flatten