Adds unit tests for FLEX_GRID