Numpy testing for decomposition

kvanderwijst · Feb 13, 2020 · 8bddb7f · 8bddb7f
1 parent 2eb08fc
commit 8bddb7f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/decomposition.py b/decomposition.py
@@ -39,6 +39,11 @@ def decomposition(n, values_start, values_end):
 
                 effect[i] += contribution_level
 
+    value_start = np.prod(values_start)
+    value_end = np.prod(values_end)
+
+    np.testing.assert_almost_equal(np.sum(effect), value_end - value_start)
+
     return {
         'value_start': np.prod(values_start),
         'value_end': np.prod(values_end),