use CellTypeUtils::list_types to loop over possible cell types in tests

1 files changed, 5 insertions, 6 deletions

diff --git a/eval/src/tests/instruction/generic_cell_cast/generic_cell_cast_test.cpp b/eval/src/tests/instruction/generic_cell_cast/generic_cell_cast_test.cpp
index 6a0b63bd562..06b74eff4d5 100644
--- a/eval/src/tests/instruction/generic_cell_cast/generic_cell_cast_test.cpp
+++ b/eval/src/tests/instruction/generic_cell_cast/generic_cell_cast_test.cpp
@@ -44,13 +44,12 @@ TensorSpec perform_generic_cell_cast(const TensorSpec &a, CellType to, const Val
 
 void test_generic_cell_cast_with(const ValueBuilderFactory &factory) {
     for (const auto &layout : layouts) {
-        for (TensorSpec lhs : { layout.cpy().cells_float(),
-                                layout.cpy().cells_double() })
-        {
-            for (CellType ct : { CellType::FLOAT, CellType::DOUBLE }) {
+        for (CellType in_type: CellTypeUtils::list_types()) {
+            for (CellType out_type: CellTypeUtils::list_types()) {
+                TensorSpec lhs = layout.cpy().cells(in_type);
                 SCOPED_TRACE(fmt("\n===\nLHS: %s\n===\n", lhs.to_string().c_str()));
-                auto expect = ReferenceOperations::cell_cast(lhs, ct);
-                auto actual = perform_generic_cell_cast(lhs, ct, factory);
+                auto expect = ReferenceOperations::cell_cast(lhs, out_type);
+                auto actual = perform_generic_cell_cast(lhs, out_type, factory);
                 EXPECT_EQ(actual, expect);
             }
         }