Mercurial > hg > CbC > CbC_llvm
diff mlir/test/APITest.h @ 150:1d019706d866
LLVM10
author | anatofuz |
---|---|
date | Thu, 13 Feb 2020 15:10:13 +0900 (2020-02-13) |
parents | |
children | c4bab56944e8 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mlir/test/APITest.h Thu Feb 13 15:10:13 2020 +0900 @@ -0,0 +1,63 @@ +//===- Test.h - Simple macros for API unit tests ----------------*- C++ -*-===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// +// +// This file define simple macros for declaring test functions and running them. +// The actual checking must be performed on the outputs with FileCheck. +// +//===----------------------------------------------------------------------===// + +#ifndef MLIR_TEST_TEST_H_ +#define MLIR_TEST_TEST_H_ + +#include <functional> +#include <vector> + +namespace test_detail { +// Returns a mutable list of known test functions. Used internally by test +// macros to add and run tests. This function is static to ensure it creates a +// new list in each test file. +static std::vector<std::function<void()>> &tests() { + static std::vector<std::function<void()>> list; + return list; +} + +// Test registration class. Used internally by test macros to register tests +// during static allocation. +struct TestRegistration { + explicit TestRegistration(std::function<void()> func) { + test_detail::tests().push_back(func); + } +}; +} // end namespace test_detail + +/// Declares a test function with the given name and adds it to the list of +/// known tests. The body of the function must follow immediately. Example: +/// +/// TEST_FUNC(mytest) { +/// // CHECK: expected-output-here +/// emitSomethingToStdOut(); +/// } +/// +#define TEST_FUNC(name) \ + void name(); \ + static test_detail::TestRegistration name##Registration(name); \ + void name() + +/// Runs all registered tests. Example: +/// +/// int main() { +/// RUN_TESTS(); +/// return 0; +/// } +#define RUN_TESTS \ + []() { \ + for (auto f : test_detail::tests()) \ + f(); \ + } + +#endif // MLIR_TEST_TEST_H_