deterministic and fuzzing testing of variance

hands_on/numerical_fuzzing/test_var.py

@@ -0,0 +1,20 @@
+import numpy as np
+from math import isclose
+
+def test_mean_deterministic():
+    x = np.array([-2.0, 2.0, 6.0])
+    expected = float(32/3)
+    print(np.var(x))
+    assert isclose(np.var(x), expected)
+
+def test_mean_fuzzing():
+    rand_state = np.random.RandomState(1333)
+
+    N, D = 100000, 5
+    # Goal means: [0.1 ,  0.45,  0.8 ,  1.15,  1.5]
+    expected = np.linspace(0.1, 1.5, D)
+
+    # Generate random, D-dimensional data with the desired mean
+    x = rand_state.randn(N, D) * np.sqrt(expected)
+    vars = np.var(x, axis=0)
+    np.testing.assert_allclose(vars, expected, rtol=1e-2)

hands_on/pyanno_voting/pyanno/tests/test_voting.py

@@ -3,6 +3,8 @@
 from pyanno import voting
 from pyanno.voting import MISSING_VALUE as MV
 
+from math import isclose
+
 
 def test_labels_count():
     annotations = [
@@ -41,3 +43,20 @@ def test_majority_vote_empty_item():
     expected = [1, MV, 2]
     result = voting.majority_vote(annotations)
     assert result == expected
+
+def test_labels_frequency():
+    matrix = [
+        [1, 2, 2, -1],
+        [2, 2, 2, 2],
+        [1, 1, 3, 3],
+        [1, 3, 3, 2],
+        [-1, 2, 3, 1],
+        [-1, -1, -1, 3],
+    ]
+    result = voting.labels_frequency(matrix, 4)
+
+    assert np.all([res != None for res in result])
+    assert len(result) == 4
+    assert np.all(voting.labels_frequency([[-1, -1, -1, -1],[-1, -1, -1, -1]], 4) == np.zeros(4))
+    assert np.all([i >= 0 and i <= 1 for i in result])
+    assert isclose(np.sum(result),1) or isclose(np.sum(result), 0,abs_tol=1e-12)

hands_on/pyanno_voting/pyanno/voting.py

@@ -100,3 +100,20 @@ def labels_frequency(annotations, nclasses):
         freq[k] is the frequency of elements of class k in `annotations`, i.e.
         their count over the number of total of observed (non-missing) elements
     """
+    annotations_array = np.ravel(annotations)
+    result = np.zeros(nclasses)
+
+    dim = 0
+    for number in annotations_array:
+        if number != -1:
+            dim = dim + 1
+    if dim !=0:
+        for cl in np.arange(nclasses):
+            aux = 0
+            for anot in annotations_array:
+                if cl == anot:
+                    aux = aux + 1
+            result[cl] = aux / dim
+        return(result)
+    else:
+        return(0)