flang/unittests/Evaluate/testing.cpp

   1 #include "testing.h"
   2 #include "llvm/Support/raw_ostream.h"
   3 #include <cstdarg>
   4 #include <cstdio>
   5 #include <cstdlib>
   6
   7 namespace testing {
   8
   9 namespace {
  10 int passes{0};
  11 int failures{0};
  12 } // namespace
  13
  14 static void BitBucket(const char *, ...) {}
  15
  16 static void PrintFailureDetails(const char *format, ...) {
  17   va_list ap;
  18   va_start(ap, format);
  19   fputs("\t", stderr);
  20   vfprintf(stderr, format, ap);
  21   va_end(ap);
  22   fputc('\n', stderr);
  23 }
  24
  25 FailureDetailPrinter Test(
  26     const char *file, int line, const char *predicate, bool pass) {
  27   if (pass) {
  28     ++passes;
  29     return BitBucket;
  30   } else {
  31     ++failures;
  32     fprintf(stderr, "%s:%d: FAIL: %s\n", file, line, predicate);
  33     return PrintFailureDetails;
  34   }
  35 }
  36
  37 FailureDetailPrinter Match(const char *file, int line, std::uint64_t want,
  38     const char *gots, std::uint64_t got) {
  39   if (want == got) {
  40     ++passes;
  41     return BitBucket;
  42   } else {
  43     ++failures;
  44     fprintf(stderr, "%s:%d: FAIL: %s == 0x%jx, not 0x%jx\n", file, line, gots,
  45         static_cast<std::uintmax_t>(got), static_cast<std::uintmax_t>(want));
  46     return PrintFailureDetails;
  47   }
  48 }
  49
  50 FailureDetailPrinter Match(const char *file, int line, const char *want,
  51     const char *gots, const std::string &got) {
  52   if (want == got) {
  53     ++passes;
  54     return BitBucket;
  55   } else {
  56     ++failures;
  57     fprintf(stderr, "%s:%d: FAIL: %s == \"%s\", not \"%s\"\n", file, line, gots,
  58         got.data(), want);
  59     return PrintFailureDetails;
  60   }
  61 }
  62
  63 FailureDetailPrinter Match(const char *file, int line, const std::string &want,
  64     const char *gots, const std::string &got) {
  65   return Match(file, line, want.data(), gots, got);
  66 }
  67
  68 FailureDetailPrinter Compare(const char *file, int line, const char *xs,
  69     const char *rel, const char *ys, std::uint64_t x, std::uint64_t y) {
  70   while (*rel == ' ') {
  71     ++rel;
  72   }
  73   bool pass{false};
  74   if (*rel == '<') {
  75     if (rel[1] == '=') {
  76       pass = x <= y;
  77     } else {
  78       pass = x < y;
  79     }
  80   } else if (*rel == '>') {
  81     if (rel[1] == '=') {
  82       pass = x >= y;
  83     } else {
  84       pass = x > y;
  85     }
  86   } else if (*rel == '=') {
  87     pass = x == y;
  88   } else if (*rel == '!') {
  89     pass = x != y;
  90   }
  91   if (pass) {
  92     ++passes;
  93     return BitBucket;
  94   } else {
  95     ++failures;
  96     fprintf(stderr, "%s:%d: FAIL: %s[0x%jx] %s %s[0x%jx]\n", file, line, xs,
  97         static_cast<std::uintmax_t>(x), rel, ys,
  98         static_cast<std::uintmax_t>(y));
  99     return PrintFailureDetails;
 100   }
 101 }
 102
 103 int Complete() {
 104   if (failures == 0) {
 105     if (passes == 1) {
 106       llvm::outs() << "single test PASSES\n";
 107     } else {
 108       llvm::outs() << "all " << passes << " tests PASS\n";
 109     }
 110     passes = 0;
 111     return EXIT_SUCCESS;
 112   } else {
 113     if (passes == 1) {
 114       llvm::errs() << "1 test passes, ";
 115     } else {
 116       llvm::errs() << passes << " tests pass, ";
 117     }
 118     if (failures == 1) {
 119       llvm::errs() << "1 test FAILS\n";
 120     } else {
 121       llvm::errs() << failures << " tests FAIL\n";
 122     }
 123     passes = failures = 0;
 124     return EXIT_FAILURE;
 125   }
 126 }
 127 } // namespace testing