test cases

Author: aajayi-21

diffpy/snmf/subroutines.py

@@ -110,6 +110,7 @@ def construct_component_matrix(components):
         component_matrix[i] = component.iq
     return component_matrix
 
+
 def construct_weight_matrix(components):
     """Constructs the weights matrix
 

diffpy/snmf/tests/test_subroutines.py

@@ -2,7 +2,7 @@
 import numpy as np
 from diffpy.snmf.containers import ComponentSignal
 from diffpy.snmf.subroutines import objective_function, get_stretched_component, reconstruct_data, get_residual_matrix, \
-    update_weights_matrix, initialize_arrays, lift_data, initialize_components, construct_stretching_matrix, construct_component_matrix
+    update_weights_matrix, initialize_arrays, lift_data, initialize_components, construct_stretching_matrix, construct_component_matrix, construct_weight_matrix
 
 to = [
     ([[[1, 2], [3, 4]], [[5, 6], [7, 8]], 1e11, [[1, 2], [3, 4]], [[1, 2], [3, 4]], 1], 2.574e14),
@@ -207,6 +207,19 @@ def test_construct_component_matrix(tccm):
     for component in tccm:
         np.testing.assert_allclose(actual[component.id], component.iq)
 
-
-def test_construct_weight_matrix():
-    assert False
+tcwm = [
+        ([ComponentSignal([0,.25,.5,.75,1],20,0)]),
+        ([ComponentSignal([0,.25,.5,.75,1],0,0)]),
+        ([ComponentSignal([0,.25,.5,.75,1],20,0),ComponentSignal([0,.25,.5,.75,1],20,1),ComponentSignal([0,.25,.5,.75,1],20,2)]),
+        ([ComponentSignal([0, .25, .5, .75, 1], 20, 0), ComponentSignal([0, .25, .5, .75, 1], 20, 1),
+          ComponentSignal([0, .25, .5, .75, 1], 20, 2)]),
+        ([ComponentSignal([0, .25, .5, .75, 1], 20, 0), ComponentSignal([0, .25, .5, 2.75, 1], 20, 1),
+          ComponentSignal([0, .25, .5, .75, 1], 20, 2)]),
+        ([ComponentSignal([.25], 20, 0), ComponentSignal([.25], 20, 1), ComponentSignal([.25], 20, 2)]),
+        ([ComponentSignal([0, .25, .5, .75, 1], 20, 0), ComponentSignal([0, .25, .5, .75, 1], 20, 1)]),
+]
+@pytest.mark.parametrize('tcwm',tcwm)
+def test_construct_weight_matrix(tcwm):
+    actual = construct_weight_matrix(tcwm)
+    for component in tcwm:
+        np.testing.assert_allclose(actual[component.id], component.weights)