Files
kompute/test/TestTensor.cpp
Alejandro Saucedo 63e220a8a4 Reformat
2021-02-28 16:02:37 +00:00

14 lines
294 B
C++

#include "gtest/gtest.h"
#include "kompute/Kompute.hpp"
TEST(TestTensor, ConstructorData)
{
kp::Manager mgr;
std::vector<float> vec{ 0, 1, 2 };
std::shared_ptr<kp::Tensor> tensor = mgr.tensor(vec);
EXPECT_EQ(tensor->size(), vec.size());
EXPECT_EQ(tensor->data(), vec);
}