axis_variable_test.cpp 5.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162
  1. // Copyright 2015-2017 Hans Dembinski
  2. //
  3. // Distributed under the Boost Software License, Version 1.0.
  4. // (See accompanying file LICENSE_1_0.txt
  5. // or copy at http://www.boost.org/LICENSE_1_0.txt)
  6. #include <boost/core/lightweight_test.hpp>
  7. #include <boost/histogram/axis/ostream.hpp>
  8. #include <boost/histogram/axis/variable.hpp>
  9. #include <limits>
  10. #include <sstream>
  11. #include <type_traits>
  12. #include <vector>
  13. #include "is_close.hpp"
  14. #include "std_ostream.hpp"
  15. #include "throw_exception.hpp"
  16. #include "utility_axis.hpp"
  17. #include "utility_str.hpp"
  18. using namespace boost::histogram;
  19. int main() {
  20. BOOST_TEST(std::is_nothrow_move_assignable<axis::variable<>>::value);
  21. BOOST_TEST(std::is_nothrow_move_constructible<axis::variable<>>::value);
  22. // bad_ctors
  23. {
  24. BOOST_TEST_THROWS(axis::variable<>(std::vector<double>{}), std::invalid_argument);
  25. BOOST_TEST_THROWS(axis::variable<>({1.0}), std::invalid_argument);
  26. BOOST_TEST_THROWS(axis::variable<>({1.0, 1.0}), std::invalid_argument);
  27. BOOST_TEST_THROWS(axis::variable<>({1.0, -1.0}), std::invalid_argument);
  28. }
  29. // axis::variable
  30. {
  31. axis::variable<> a{{-1, 0, 1}, "foo"};
  32. BOOST_TEST_EQ(a.size(), 2);
  33. BOOST_TEST_EQ(a.metadata(), "foo");
  34. BOOST_TEST_EQ(static_cast<const axis::variable<>&>(a).metadata(), "foo");
  35. a.metadata() = "bar";
  36. BOOST_TEST_EQ(static_cast<const axis::variable<>&>(a).metadata(), "bar");
  37. BOOST_TEST_EQ(a.bin(-1).lower(), -std::numeric_limits<double>::infinity());
  38. BOOST_TEST_EQ(a.bin(a.size()).upper(), std::numeric_limits<double>::infinity());
  39. BOOST_TEST_EQ(a.value(0), -1);
  40. BOOST_TEST_EQ(a.value(0.5), -0.5);
  41. BOOST_TEST_EQ(a.value(1), 0);
  42. BOOST_TEST_EQ(a.value(1.5), 0.5);
  43. BOOST_TEST_EQ(a.value(2), 1);
  44. BOOST_TEST_EQ(a.index(-10), -1);
  45. BOOST_TEST_EQ(a.index(-1), 0);
  46. BOOST_TEST_EQ(a.index(0), 1);
  47. BOOST_TEST_EQ(a.index(1), 2);
  48. BOOST_TEST_EQ(a.index(10), 2);
  49. BOOST_TEST_EQ(a.index(-std::numeric_limits<double>::infinity()), -1);
  50. BOOST_TEST_EQ(a.index(std::numeric_limits<double>::infinity()), 2);
  51. BOOST_TEST_EQ(a.index(std::numeric_limits<double>::quiet_NaN()), 2);
  52. BOOST_TEST_EQ(str(a),
  53. "variable(-1, 0, 1, metadata=\"bar\", options=underflow | overflow)");
  54. axis::variable<> b;
  55. BOOST_TEST_NE(a, b);
  56. b = a;
  57. BOOST_TEST_EQ(a, b);
  58. axis::variable<> c = std::move(b);
  59. BOOST_TEST_EQ(c, a);
  60. axis::variable<> d;
  61. BOOST_TEST_NE(c, d);
  62. d = std::move(c);
  63. BOOST_TEST_EQ(d, a);
  64. axis::variable<> e{-2, 0, 2};
  65. BOOST_TEST_NE(a, e);
  66. }
  67. // axis::variable circular
  68. {
  69. axis::variable<double, axis::null_type, axis::option::circular_t> a{-1, 1, 2};
  70. BOOST_TEST_EQ(a.value(-2), -4);
  71. BOOST_TEST_EQ(a.value(-1), -2);
  72. BOOST_TEST_EQ(a.value(0), -1);
  73. BOOST_TEST_EQ(a.value(1), 1);
  74. BOOST_TEST_EQ(a.value(2), 2);
  75. BOOST_TEST_EQ(a.value(3), 4);
  76. BOOST_TEST_EQ(a.value(4), 5);
  77. BOOST_TEST_EQ(a.index(-3), 0); // -3 + 3 = 0
  78. BOOST_TEST_EQ(a.index(-2), 1); // -2 + 3 = 1
  79. BOOST_TEST_EQ(a.index(-1), 0);
  80. BOOST_TEST_EQ(a.index(0), 0);
  81. BOOST_TEST_EQ(a.index(1), 1);
  82. BOOST_TEST_EQ(a.index(2), 0);
  83. BOOST_TEST_EQ(a.index(3), 0); // 3 - 3 = 0
  84. BOOST_TEST_EQ(a.index(4), 1); // 4 - 3 = 1
  85. }
  86. // axis::regular with growth
  87. {
  88. axis::variable<double, axis::null_type, axis::option::growth_t> a{0, 1};
  89. BOOST_TEST_EQ(a.size(), 1);
  90. BOOST_TEST_EQ(a.update(0), std::make_pair(0, 0));
  91. BOOST_TEST_EQ(a.size(), 1);
  92. BOOST_TEST_EQ(a.update(1.1), std::make_pair(1, -1));
  93. BOOST_TEST_EQ(a.size(), 2);
  94. BOOST_TEST_EQ(a.value(0), 0);
  95. BOOST_TEST_EQ(a.value(1), 1);
  96. BOOST_TEST_EQ(a.value(2), 1.5);
  97. BOOST_TEST_EQ(a.update(-0.1), std::make_pair(0, 1));
  98. BOOST_TEST_EQ(a.value(0), -0.5);
  99. BOOST_TEST_EQ(a.size(), 3);
  100. BOOST_TEST_EQ(a.update(10), std::make_pair(3, -1));
  101. BOOST_TEST_EQ(a.size(), 4);
  102. BOOST_TEST_IS_CLOSE(a.value(4), 10, 1e-9);
  103. BOOST_TEST_EQ(a.update(-10), std::make_pair(0, 1));
  104. BOOST_TEST_EQ(a.size(), 5);
  105. BOOST_TEST_IS_CLOSE(a.value(0), -10, 1e-9);
  106. BOOST_TEST_EQ(a.update(-std::numeric_limits<double>::infinity()),
  107. std::make_pair(-1, 0));
  108. BOOST_TEST_EQ(a.update(std::numeric_limits<double>::infinity()),
  109. std::make_pair(a.size(), 0));
  110. BOOST_TEST_EQ(a.update(std::numeric_limits<double>::quiet_NaN()),
  111. std::make_pair(a.size(), 0));
  112. }
  113. // iterators
  114. {
  115. test_axis_iterator(axis::variable<>{1, 2, 3}, 0, 2);
  116. test_axis_iterator(
  117. axis::variable<double, axis::null_type, axis::option::circular_t>{1, 2, 3}, 0, 2);
  118. }
  119. // shrink and rebin
  120. {
  121. using A = axis::variable<>;
  122. auto a = A({0, 1, 2, 3, 4, 5});
  123. auto b = A(a, 1, 4, 1);
  124. BOOST_TEST_EQ(b.size(), 3);
  125. BOOST_TEST_EQ(b.value(0), 1);
  126. BOOST_TEST_EQ(b.value(3), 4);
  127. auto c = A(a, 0, 4, 2);
  128. BOOST_TEST_EQ(c.size(), 2);
  129. BOOST_TEST_EQ(c.value(0), 0);
  130. BOOST_TEST_EQ(c.value(2), 4);
  131. auto e = A(a, 1, 5, 2);
  132. BOOST_TEST_EQ(e.size(), 2);
  133. BOOST_TEST_EQ(e.value(0), 1);
  134. BOOST_TEST_EQ(e.value(2), 5);
  135. }
  136. // shrink and rebin with circular option
  137. {
  138. using A = axis::variable<double, axis::null_type, axis::option::circular_t>;
  139. auto a = A({1, 2, 3, 4, 5});
  140. BOOST_TEST_THROWS(A(a, 1, 4, 1), std::invalid_argument);
  141. BOOST_TEST_THROWS(A(a, 0, 3, 1), std::invalid_argument);
  142. auto b = A(a, 0, 4, 2);
  143. BOOST_TEST_EQ(b.size(), 2);
  144. BOOST_TEST_EQ(b.value(0), 1);
  145. BOOST_TEST_EQ(b.value(2), 5);
  146. }
  147. return boost::report_errors();
  148. }