Change doc and test to the more general iterable input

Author: BastienTr

commpy/sequences.py

@@ -29,11 +29,11 @@ def pnsequence(pn_order, pn_seed, pn_mask, seq_length):
     pn_order : int
         Number of delay elements used in the LFSR.
 
-    pn_seed : string containing 0's and 1's
+    pn_seed : iterable providing 0's and 1's
         Seed for the initialization of the LFSR delay elements.
         The length of this string must be equal to 'pn_order'.
 
-    pn_mask : string containing 0's and 1's
+    pn_mask : iterable providing 0's and 1's
         Mask representing which delay elements contribute to the feedback
         in the LFSR. The length of this string must be equal to 'pn_order'.
 

commpy/tests/test_sequences.py

@@ -15,7 +15,9 @@ def test_pnsequence():
         pnsequence(4, '0011', '110', 2 ** 4 - 1)
 
     # Test output with
-    assert_equal(pnsequence(4, '0011', '1101', 7), array((1, 1, 0, 0, 1, 0, 1), int),
+    assert_equal(pnsequence(4, '0011', [1, 1, 0, 1], 7), array((1, 1, 0, 0, 1, 0, 1), int),
+                 err_msg='Pseudo-noise sequence is not the one expected.')
+    assert_equal(pnsequence(4, (0, 0, 1, 1), array((1, 1, 0, 1)), 7), array((1, 1, 0, 0, 1, 0, 1), int),
                  err_msg='Pseudo-noise sequence is not the one expected.')
 
 if __name__ == "__main__":