File: TestInitializeFinalize.cpp

package info (click to toggle)
kokkos 4.7.01-2
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 16,636 kB
  • sloc: cpp: 223,676; sh: 2,446; makefile: 2,437; python: 91; fortran: 4; ansic: 2
file content (115 lines) | stat: -rw-r--r-- 3,144 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
//@HEADER
// ************************************************************************
//
//                        Kokkos v. 4.0
//       Copyright (2022) National Technology & Engineering
//               Solutions of Sandia, LLC (NTESS).
//
// Under the terms of Contract DE-NA0003525 with NTESS,
// the U.S. Government retains certain rights in this software.
//
// Part of Kokkos, under the Apache License v2.0 with LLVM Exceptions.
// See https://kokkos.org/LICENSE for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//@HEADER

#include <Kokkos_Core.hpp>

#include <gtest/gtest.h>

#include <cstdlib>

#include "KokkosExecutionEnvironmentNeverInitializedFixture.hpp"

namespace {

using InitializeFinalize_DeathTest = KokkosExecutionEnvironmentNeverInitialized;

TEST_F(InitializeFinalize_DeathTest, initialize) {
  ::testing::FLAGS_gtest_death_test_style = "threadsafe";

  EXPECT_EXIT(
      {
        Kokkos::initialize();
        Kokkos::finalize();
        std::exit(EXIT_SUCCESS);
      },
      ::testing::ExitedWithCode(EXIT_SUCCESS), "");

  EXPECT_DEATH(
      {
        Kokkos::initialize();
        Kokkos::initialize();
      },
      "Error: Kokkos::initialize\\(\\) has already been called. Kokkos can be "
      "initialized at most once\\.");

  EXPECT_DEATH(
      {
        Kokkos::initialize();
        Kokkos::finalize();
        Kokkos::initialize();
      },
      "Error: Kokkos::initialize\\(\\) has already been called. Kokkos can be "
      "initialized at most once\\.");
}

TEST_F(InitializeFinalize_DeathTest, finalize) {
  ::testing::FLAGS_gtest_death_test_style = "threadsafe";

  EXPECT_DEATH(
      { Kokkos::finalize(); },
      "Error: Kokkos::finalize\\(\\) may only be called after Kokkos has "
      "been initialized\\.");

  EXPECT_EXIT(
      {
        Kokkos::initialize();
        Kokkos::finalize();
        std::exit(EXIT_SUCCESS);
      },
      ::testing::ExitedWithCode(EXIT_SUCCESS), "");

  EXPECT_DEATH(
      {
        Kokkos::initialize();
        Kokkos::finalize();
        Kokkos::finalize();
      },
      "Error: Kokkos::finalize\\(\\) has already been called\\.");
}

TEST_F(InitializeFinalize_DeathTest, is_initialized) {
  ::testing::FLAGS_gtest_death_test_style = "threadsafe";

  EXPECT_EXIT(
      {
        bool success = true;
        success &= !Kokkos::is_initialized();
        Kokkos::initialize();
        success &= Kokkos::is_initialized();
        Kokkos::finalize();
        success &= !Kokkos::is_initialized();
        std::exit(success ? EXIT_SUCCESS : EXIT_FAILURE);
      },
      ::testing::ExitedWithCode(EXIT_SUCCESS), "");
}

TEST_F(InitializeFinalize_DeathTest, is_finalized) {
  ::testing::FLAGS_gtest_death_test_style = "threadsafe";

  EXPECT_EXIT(
      {
        bool success = true;
        success &= !Kokkos::is_finalized();
        Kokkos::initialize();
        success &= !Kokkos::is_finalized();
        Kokkos::finalize();
        success &= Kokkos::is_finalized();
        std::exit(success ? EXIT_SUCCESS : EXIT_FAILURE);
      },
      ::testing::ExitedWithCode(EXIT_SUCCESS), "");
}

}  // namespace