Fix ndarray doctest

scbirlab · Oct 11, 2024 · da452f4 · da452f4
1 parent 582802a
commit da452f4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/schemist/features.py b/schemist/features.py
@@ -190,13 +190,13 @@ def calculate_fingerprints(
     Examples
     --------
     >>> bits, validity = calculate_fingerprints(strings='CCC')
-    >>> bits
-    array(['80;294;1057;1344'], dtype='<U16')
+    >>> list(bits)
+    ['80;294;1057;1344']
     >>> sum(validity)  # doctest: +NORMALIZE_WHITESPACE
     1 
     >>> bits, validity = calculate_fingerprints(strings=['CCC', 'CCCO'])
-    >>> bits
-    array(['80;294;1057;1344', '80;222;294;473;794;807;1057;1277'], dtype='<U16')
+    >>> list(bits)
+    ['80;294;1057;1344', '80;222;294;473;794;807;1057;1277']
     >>> sum(validity)  # doctest: +NORMALIZE_WHITESPACE
     2
     >>> np.sum(calculate_fingerprints(strings=['CCC', 'CCCO'], on_bits=False)[0], axis=-1)