count Nan fix

apache · Aug 20, 2023 · 6b2e514 · 6b2e514
1 parent 6a47608
commit 6b2e514
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/test/java/org/apache/sysds/test/component/compress/util/CountMapTest.java b/src/test/java/org/apache/sysds/test/component/compress/util/CountMapTest.java
@@ -217,9 +217,15 @@ public void getDictionary() {
 			assertEquals((double) i, d[i], 0.0);
 	}
 
-	@Test(expected = NullPointerException.class)
+	@Test()
 	public void getNan() {
 		m.increment(Double.NaN);
-		m.getId(Double.NaN);
+		assertEquals(0, m.getId(Double.NaN));
+		m.increment(Double.NaN);
+		assertEquals(0, m.getId(Double.NaN));
+		m.increment(1.0);
+		assertEquals(0, m.getId(Double.NaN));
+		m.increment(Double.NaN);
+		assertEquals(0, m.getId(Double.NaN));
 	}
 }