Added NDArray, did some light header and CMake restructuring
This commit is contained in:
@@ -3,9 +3,9 @@
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
#include <string>
|
||||
#include <NBCore/DataTypes.hpp>
|
||||
#include <NBCore/TypeTraits.hpp>
|
||||
#include <NBCore/StringUtils.hpp>
|
||||
#include <NBCore/TypeTraits.hpp>
|
||||
#include <NBCore/Utils.hpp>
|
||||
|
||||
namespace nb {
|
||||
|
||||
@@ -107,22 +107,4 @@ TEST(UtilsTest, TestIsDetected) {
|
||||
ASSERT_TRUE(ret);
|
||||
}
|
||||
|
||||
TEST(DataTypesTests, TestNDArray) {
|
||||
float* data_f = new float[9] {0,1,2,3,4,5,6,7,8};
|
||||
nb::NDArray<float, 2> arr({3, 3}, data_f);
|
||||
ASSERT_FLOAT_EQ(data_f[4], arr.at({1,1}));
|
||||
arr.at({1,1}) = 0.5;
|
||||
ASSERT_FLOAT_EQ(0.5, arr.at({1,1}));
|
||||
delete[] data_f;
|
||||
|
||||
unsigned int* data_ui = new unsigned int[8] {0,1,2,3,4,5,6,7};
|
||||
nb::NDArray<unsigned int, 3> arr3d({2,2,2}, data_ui);
|
||||
ASSERT_EQ(data_ui[5], arr3d.at({1,0,1}));
|
||||
arr3d.at({0,1,1}) = 23;
|
||||
ASSERT_EQ(23, arr3d.at({0,1,1}));
|
||||
delete[] data_ui;
|
||||
|
||||
|
||||
}
|
||||
|
||||
} // namespace nb
|
||||
|
||||
Reference in New Issue
Block a user