Add tests for format module