1 //===- unittests/IR/ModuleTest.cpp - Module unit tests --------------------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 #include "llvm/IR/Module.h"
11 #include "llvm/IR/GlobalVariable.h"
12 #include "llvm/Support/RandomNumberGenerator.h"
13 #include "gtest/gtest.h"
21 bool sortByName(const GlobalVariable &L, const GlobalVariable &R) {
22 return L.getName() < R.getName();
25 bool sortByNameReverse(const GlobalVariable &L, const GlobalVariable &R) {
26 return sortByName(R, L);
29 TEST(ModuleTest, sortGlobalsByName) {
31 for (auto compare : {&sortByName, &sortByNameReverse}) {
32 Module M("M", Context);
33 Type *T = Type::getInt8Ty(Context);
34 GlobalValue::LinkageTypes L = GlobalValue::ExternalLinkage;
35 (void)new GlobalVariable(M, T, false, L, nullptr, "A");
36 (void)new GlobalVariable(M, T, false, L, nullptr, "F");
37 (void)new GlobalVariable(M, T, false, L, nullptr, "G");
38 (void)new GlobalVariable(M, T, false, L, nullptr, "E");
39 (void)new GlobalVariable(M, T, false, L, nullptr, "B");
40 (void)new GlobalVariable(M, T, false, L, nullptr, "H");
41 (void)new GlobalVariable(M, T, false, L, nullptr, "C");
42 (void)new GlobalVariable(M, T, false, L, nullptr, "D");
44 // Sort the globals by name.
45 EXPECT_FALSE(std::is_sorted(M.global_begin(), M.global_end(), compare));
46 M.getGlobalList().sort(compare);
47 EXPECT_TRUE(std::is_sorted(M.global_begin(), M.global_end(), compare));
51 TEST(ModuleTest, randomNumberGenerator) {
54 struct DummyPass : ModulePass {
55 DummyPass() : ModulePass(ID) {}
56 bool runOnModule(Module &) { return true; }
59 Module M("R", Context);
61 std::uniform_int_distribution<int> dist;
62 const size_t NBCheck = 10;
64 std::array<int, NBCheck> RandomStreams[2];
65 for (auto &RandomStream : RandomStreams) {
66 std::unique_ptr<RandomNumberGenerator> RNG = M.createRNG(&DP);
67 std::generate(RandomStream.begin(), RandomStream.end(),
68 [&]() { return dist(*RNG); });
71 EXPECT_TRUE(std::equal(RandomStreams[0].begin(), RandomStreams[0].end(),
72 RandomStreams[1].begin()));