Merge pull request #44 from bmcfee/pump-fields

added fields property to Pump
bmcfee · Mar 16, 2017 · b1976b5 · b1976b5
2 parents ea31faa + 52e53ed
commit b1976b5
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/pumpp/core.py b/pumpp/core.py
@@ -155,3 +155,11 @@ def sampler(self, n_samples, duration):
         '''
 
         return Sampler(n_samples, duration, *self.ops)
+
+    @property
+    def fields(self):
+        out = dict()
+        for op in self.ops:
+            out.update(**op.fields)
+
+        return out
diff --git a/tests/test_core.py b/tests/test_core.py
@@ -93,6 +93,12 @@ def test_pump(audio_f, jam, sr, hop_length):
     for key in data1:
         assert np.allclose(data1[key], data2[key])
 
+    fields = dict()
+    for op in ops:
+        fields.update(**op.fields)
+
+    assert pump.fields == fields
+
 
 @pytest.mark.parametrize('audio_f', ['tests/data/test.ogg'])
 def test_pump_empty(audio_f, jam, sr, hop_length):