added instantiation tests to get covereddd

bmcfee · Oct 8, 2019 · 03216e0 · 03216e0
1 parent 566cd55
commit 03216e0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/test_core.py b/tests/test_core.py
@@ -213,6 +213,10 @@ def test_pump_layers(sr, hop_length):
         for d1, d2 in zip(L1[k].shape, L2[k].shape):
             assert str(d1) == str(d2)
 
+    # test other input layers
+    P.layers('tf.keras')
+    P.layers('tf')
+
 
 def test_pump_str(sr, hop_length):