fmt/test/custom-formatter-test.cc

47 lines
1.4 KiB
C++
Raw Normal View History

2018-03-04 17:16:51 +00:00
// Formatting library for C++ - custom argument formatter tests
//
// Copyright (c) 2012 - present, Victor Zverovich
// All rights reserved.
//
// For the license information refer to format.h.
#include "fmt/format.h"
#include "gtest-extra.h"
// A custom argument formatter that doesn't print `-` for floating-point values
// rounded to 0.
2018-03-04 17:16:51 +00:00
class custom_arg_formatter :
2018-01-15 16:22:31 +00:00
public fmt::arg_formatter<fmt::back_insert_range<fmt::internal::buffer>> {
public:
typedef fmt::back_insert_range<fmt::internal::buffer> range;
typedef fmt::arg_formatter<range> base;
2018-01-14 15:19:23 +00:00
2018-03-04 17:16:51 +00:00
custom_arg_formatter(fmt::context &ctx, fmt::format_specs &s)
2018-01-15 16:22:31 +00:00
: base(ctx, s) {}
2018-01-14 15:19:23 +00:00
using base::operator();
2016-11-20 16:47:24 +00:00
iterator operator()(double value) {
if (round(value * pow(10, spec().precision())) == 0)
value = 0;
return base::operator()(value);
}
};
2017-12-03 15:32:04 +00:00
std::string custom_vformat(fmt::string_view format_str, fmt::format_args args) {
fmt::memory_buffer buffer;
// Pass custom argument formatter as a template arg to vwrite.
2018-03-04 17:16:51 +00:00
fmt::vformat_to<custom_arg_formatter>(buffer, format_str, args);
return std::string(buffer.data(), buffer.size());
}
2016-08-27 00:23:13 +00:00
template <typename... Args>
std::string custom_format(const char *format_str, const Args & ... args) {
2017-02-05 14:41:39 +00:00
auto va = fmt::make_args(args...);
2016-08-27 14:57:48 +00:00
return custom_vformat(format_str, va);
2016-08-27 00:23:13 +00:00
}
TEST(CustomFormatterTest, Format) {
EXPECT_EQ("0.00", custom_format("{:.2f}", -.00001));
}