assert.h 4.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114
  1. #include <iostream>
  2. #include <cstdlib>
  3. #define equals(x,y) assertEquals(y, x, __LINE__)
  4. #define equalsNot(x,y) assertEqualsNot(y, x, __LINE__)
  5. template < typename X, typename Y >
  6. void OutputAssertEqual(X expect, char const* compare, Y get, unsigned long const &line) {
  7. std::cerr << "Test FAILED: »" << expect << "« " << compare << " »" << get << "« at line " << line << std::endl;
  8. std::exit(EXIT_FAILURE);
  9. }
  10. template < typename X, typename Y >
  11. void assertEquals(X expect, Y get, unsigned long const &line) {
  12. if (expect == get)
  13. return;
  14. OutputAssertEqual(expect, "==", get, line);
  15. }
  16. template < typename X, typename Y >
  17. void assertEqualsNot(X expect, Y get, unsigned long const &line) {
  18. if (expect != get)
  19. return;
  20. OutputAssertEqual(expect, "!=", get, line);
  21. }
  22. void assertEquals(unsigned int const &expect, int const &get, unsigned long const &line) {
  23. if (get < 0)
  24. OutputAssertEqual(expect, "==", get, line);
  25. assertEquals<unsigned int const&, unsigned int const&>(expect, get, line);
  26. }
  27. void assertEquals(int const &expect, unsigned int const &get, unsigned long const &line) {
  28. if (expect < 0)
  29. OutputAssertEqual(expect, "==", get, line);
  30. assertEquals<unsigned int const&, unsigned int const&>(expect, get, line);
  31. }
  32. void assertEquals(unsigned long const &expect, int const &get, unsigned long const &line) {
  33. if (get < 0)
  34. OutputAssertEqual(expect, "==", get, line);
  35. assertEquals<unsigned long const&, unsigned long const&>(expect, get, line);
  36. }
  37. void assertEquals(int const &expect, unsigned long const &get, unsigned long const &line) {
  38. if (expect < 0)
  39. OutputAssertEqual(expect, "==", get, line);
  40. assertEquals<unsigned long const&, unsigned long const&>(expect, get, line);
  41. }
  42. #define equalsOr2(x,y,z) assertEqualsOr2(y, z, x, __LINE__)
  43. template < typename X, typename Y >
  44. void OutputAssertEqualOr2(X expect1, X expect2, char const* compare, Y get, unsigned long const &line) {
  45. std::cerr << "Test FAILED: »" << expect1 << "« or »" << expect2 << "« " << compare << " »" << get << "« at line " << line << std::endl;
  46. }
  47. template < typename X, typename Y >
  48. void assertEqualsOr2(X expect1, X expect2, Y get, unsigned long const &line) {
  49. if (expect1 == get || expect2 == get)
  50. return;
  51. OutputAssertEqualOr2(expect1, expect2, "==", get, line);
  52. }
  53. void assertEqualsOr2(unsigned int const &expect1, unsigned int const &expect2, int const &get, unsigned long const &line) {
  54. if (get < 0)
  55. OutputAssertEqualOr2(expect1, expect2, "==", get, line);
  56. assertEqualsOr2<unsigned int const&, unsigned int const&>(expect1, expect2, get, line);
  57. }
  58. void assertEqualsOr2(int const &expect1, int const &expect2, unsigned int const &get, unsigned long const &line) {
  59. if (expect1 < 0 && expect2 < 0)
  60. OutputAssertEqualOr2(expect1, expect2, "==", get, line);
  61. assertEqualsOr2<unsigned int const&, unsigned int const&>(expect1, expect2, get, line);
  62. }
  63. #define equalsOr3(w,x,y,z) assertEqualsOr3(x, y, z, w, __LINE__)
  64. template < typename X, typename Y >
  65. void OutputAssertEqualOr3(X expect1, X expect2, X expect3, char const* compare, Y get, unsigned long const &line) {
  66. std::cerr << "Test FAILED: »" << expect1 << "« or »" << expect2 << "« or »" << expect3 << "« " << compare << " »" << get << "« at line " << line << std::endl;
  67. }
  68. template < typename X, typename Y >
  69. void assertEqualsOr3(X expect1, X expect2, X expect3, Y get, unsigned long const &line) {
  70. if (expect1 == get || expect2 == get || expect3 == get)
  71. return;
  72. OutputAssertEqualOr3(expect1, expect2, expect3, "==", get, line);
  73. }
  74. #define equalsOr4(v,w,x,y,z) assertEqualsOr4(w, x, y, z, v, __LINE__)
  75. template < typename X, typename Y >
  76. void OutputAssertEqualOr4(X expect1, X expect2, X expect3, X expect4, char const* compare, Y get, unsigned long const &line) {
  77. std::cerr << "Test FAILED: »" << expect1 << "« or »" << expect2 << "« or »" << expect3 << "« or »" << expect4 << "« " << compare << " »" << get << "« at line " << line << std::endl;
  78. }
  79. template < typename X, typename Y >
  80. void assertEqualsOr4(X expect1, X expect2, X expect3, X expect4, Y get, unsigned long const &line) {
  81. if (expect1 == get || expect2 == get || expect3 == get || expect4 == get)
  82. return;
  83. OutputAssertEqualOr4(expect1, expect2, expect3, expect4, "==", get, line);
  84. }
  85. // simple helper to quickly output a vectors
  86. template < typename X >
  87. void dumpVector(X vec) {
  88. for (typename X::const_iterator v = vec.begin();
  89. v != vec.end(); ++v)
  90. std::cout << *v << std::endl;
  91. }