fixup! add csr tests

ginkgo-project · Dec 4, 2023 · 2b04fd1 · 2b04fd1
1 parent 6ebaeae
commit 2b04fd1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/core/test/matrix/coo.cpp b/core/test/matrix/coo.cpp
@@ -253,7 +253,8 @@ TYPED_TEST(Coo, CanBeReadFromDeviceMatrixData)
     m->read(device_data);
 
     this->assert_equal_to_original_mtx(m);
-    ASSERT_EQ(device_data.get_num_elems(), m->get_num_stored_elements());
+    ASSERT_EQ(device_data.get_num_stored_elements(),
+              m->get_num_stored_elements());
     GKO_ASSERT_EQUAL_DIMENSIONS(&device_data, m);
 }
 
@@ -305,7 +306,7 @@ TYPED_TEST(Coo, CanBeReadFromMovedDeviceMatrixData)
 
     this->assert_equal_to_original_mtx(m);
     GKO_ASSERT_EQUAL_DIMENSIONS(&device_data, gko::dim<2>{});
-    ASSERT_EQ(device_data.get_num_elems(), 0);
+    ASSERT_EQ(device_data.get_num_stored_elements(), 0);
 }
 
 

diff --git a/core/test/matrix/csr.cpp b/core/test/matrix/csr.cpp
@@ -279,7 +279,8 @@ TYPED_TEST(Csr, CanBeReadFromDeviceMatrixData)
     m->read(device_data);
 
     this->assert_equal_to_original_mtx(m);
-    ASSERT_EQ(device_data.get_num_elems(), m->get_num_stored_elements());
+    ASSERT_EQ(device_data.get_num_stored_elements(),
+              m->get_num_stored_elements());
     GKO_ASSERT_EQUAL_DIMENSIONS(&device_data, m);
 }
 
@@ -333,7 +334,7 @@ TYPED_TEST(Csr, CanBeReadFromMovedDeviceMatrixData)
 
     this->assert_equal_to_original_mtx(m);
     GKO_ASSERT_EQUAL_DIMENSIONS(&device_data, gko::dim<2>{});
-    ASSERT_EQ(device_data.get_num_elems(), 0);
+    ASSERT_EQ(device_data.get_num_stored_elements(), 0);
 }