added more tests

xtensor-stack · Oct 14, 2021 · b22f8ff · b22f8ff
1 parent 66d5ebf
commit b22f8ff
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 7 deletions.
diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt
@@ -145,7 +145,7 @@ set(XTENSOR_BLAS_TESTS
     test_dot.cpp
     test_dot_extended.cpp
     test_qr.cpp
-    test_complex_float_norm.cpp
+    test_float_norm.cpp
 )
 
 add_executable(test_xtensor_blas ${XTENSOR_BLAS_TESTS} ${XTENSOR_BLAS_HEADERS} ${XTENSOR_HEADERS})

diff --git a/test/test_complex_float_norm.cpp → test/test_float_norm.cpp b/test/test_complex_float_norm.cpp → test/test_float_norm.cpp
@@ -21,12 +21,17 @@
 namespace xt
 {
 
-TEST(xblas, norm_complex_float) {
-  xt::xarray<std::complex<float>> a = {std::complex<float>(1.0f, 2.0f), std::complex<float>(3.0f, 4.0f)};
-  auto res = linalg::norm(a);
+  TEST(xblas, norm_complex_float) {
+    xt::xarray<std::complex<float>> a = {std::complex<float>(1.0f, 2.0f), std::complex<float>(3.0f, 4.0f)};
+    auto res = linalg::norm(a);
 
-  EXPECT_NEAR(res.real(), 5.4772f, 1e-3f);
-  EXPECT_NEAR(res.imag(), 0.0f, 1e-3f);
-}
+    EXPECT_NEAR(res.real(), 5.4772f, 1e-3f);
+    EXPECT_NEAR(res.imag(), 0.0f, 1e-3f);
+  }
+
+  TEST(xblas, norm_float_arange)
+  {
+    xt::linalg::norm(xt::arange<float>(15), 1); 
+  }
 
 }