crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-26 05:05:30 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Add more operator overloads in the new Tensor class (#3822)

Browse source
This commit is contained in:
Kevin Sawatzky 2026-02-19 11:03:19 -06:00 • committed by GitHub
parent 417343c920
commit efefdb17b1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 47 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

23
tests/cpp_unit_tests/test_tensor.cpp
View file

@ -350,6 +350,9 @@ TEST_CASE("Tensor element-wise arithmetic")
c = a / b;
REQUIRE(c[0] == 0.25);
c = a * b;
REQUIRE(c[0] == 4.0);
}
TEST_CASE("Tensor scalar arithmetic")
@ -378,7 +381,7 @@ TEST_CASE("Tensor scalar arithmetic")
REQUIRE(b[0] == 11.0);
}
TEST_CASE("Tensor compound addition with tensor")
TEST_CASE("Tensor compound arithmetic with tensor")
{
Tensor<double> a({3}, 0.0);
Tensor<double> b({3}, 0.0);
@ -388,6 +391,24 @@ TEST_CASE("Tensor compound addition with tensor")
REQUIRE(a[0] == 11.0);
REQUIRE(a[1] == 22.0);
REQUIRE(a[2] == 33.0);
a = {1.0, 2.0, 3.0};
b -= a;
REQUIRE(b[0] == 9.0);
REQUIRE(b[1] == 18.0);
REQUIRE(b[2] == 27.0);
b = {10.0, 20.0, 30.0};
a *= b;
REQUIRE(a[0] == 10.0);
REQUIRE(a[1] == 40.0);
REQUIRE(a[2] == 90.0);
a = {1.0, 2.0, 3.0};
b /= a;
REQUIRE(b[0] == 10.0);
REQUIRE(b[1] == 10.0);
REQUIRE(b[2] == 10.0);
}
TEST_CASE("Tensor comparison operators")