Test that fingerprint contains only valid floats

joclement · Oct 23, 2020 · 02ad6d1 · 02ad6d1
1 parent f6cdb26
commit 02ad6d1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/test_fingerprint.py b/tests/test_fingerprint.py
@@ -1,3 +1,5 @@
+import math
+
 import pandas as pd
 
 from thesis import fingerprint
@@ -7,6 +9,8 @@ def test_fingerprint_tu_graz(measurement):
     finger = fingerprint.tu_graz(measurement)
     assert type(finger) is pd.Series
     assert len(finger) == 14
+    assert all([not math.isnan(value) for value in finger])
+    assert finger.dtype == "float64"
     assert len(finger) == len(set(finger.tolist()))