1 files changed, 3 insertions, 1 deletions
diff --git a/vespajlib/src/test/java/com/yahoo/tensor/IndexedTensorTestCase.java b/vespajlib/src/test/java/com/yahoo/tensor/IndexedTensorTestCase.java
index 4bfdb53e321..1a5bae9b02c 100644
--- a/vespajlib/src/test/java/com/yahoo/tensor/IndexedTensorTestCase.java
+++ b/vespajlib/src/test/java/com/yahoo/tensor/IndexedTensorTestCase.java
@@ -146,8 +146,10 @@ public class IndexedTensorTestCase {
         // Lookup from iterator
         Map<TensorAddress, Double> cellsOfIterator = new HashMap<>();
         for (Iterator<Tensor.Cell> i = tensor.cellIterator(); i.hasNext(); ) {
-            Map.Entry<TensorAddress, Double> cell = i.next();
+            Tensor.Cell cell = i.next();
             cellsOfIterator.put(cell.getKey(), cell.getValue());
+            assertEquals(cell.getValue(), cell.getDoubleValue(), 0.00001);
+            assertEquals(cell.getValue(), cell.getFloatValue(), 0.00001);
         }
         assertEquals(tensor.size(), cellsOfIterator.size());
         for (int v = 0; v < vSize; v++)