Add a test for flattening.