From 89fce2057e7f0dc3476d86938392bf476332449e Mon Sep 17 00:00:00 2001
From: Alejandro Saucedo <axsauze@gmail.com>
Date: Sat, 11 Sep 2021 17:54:03 +0100
Subject: [PATCH] Added test for data types

Signed-off-by: Alejandro Saucedo <axsauze@gmail.com>
---
 test/TestTensor.cpp | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/test/TestTensor.cpp b/test/TestTensor.cpp
index cba3cd961..32c870a7b 100644
--- a/test/TestTensor.cpp
+++ b/test/TestTensor.cpp
@@ -10,5 +10,35 @@ TEST(TestTensor, ConstructorData)
     std::vector<float> vec{ 0, 1, 2 };
     std::shared_ptr<kp::TensorT<float>> tensor = mgr.tensor(vec);
     EXPECT_EQ(tensor->size(), vec.size());
+    EXPECT_EQ(tensor->dataTypeMemorySize(), sizeof(float));
     EXPECT_EQ(tensor->vector(), vec);
 }
+
+TEST(TestTensor, DataTypes)
+{
+    kp::Manager mgr;
+
+    {
+        std::vector<float> vec{ 0, 1, 2 };
+        std::shared_ptr<kp::TensorT<float>> tensor = mgr.tensor(vec);
+        EXPECT_EQ(tensor->dataType(), kp::Tensor::TensorDataTypes::eFloat);
+    }
+
+    {
+        std::vector<int32_t> vec{ 0, 1, 2 };
+        std::shared_ptr<kp::TensorT<int32_t>> tensor = mgr.tensorT(vec);
+        EXPECT_EQ(tensor->dataType(), kp::Tensor::TensorDataTypes::eInt);
+    }
+
+    {
+        std::vector<uint32_t> vec{ 0, 1, 2 };
+        std::shared_ptr<kp::TensorT<uint32_t>> tensor = mgr.tensorT(vec);
+        EXPECT_EQ(tensor->dataType(), kp::Tensor::TensorDataTypes::eUnsignedInt);
+    }
+
+    {
+        std::vector<double> vec{ 0, 1, 2 };
+        std::shared_ptr<kp::TensorT<double>> tensor = mgr.tensorT(vec);
+        EXPECT_EQ(tensor->dataType(), kp::Tensor::TensorDataTypes::eDouble);
+    }
+}