Reference chans from 0 to 4 in the test

Signed-off-by: Miguel Gomez <magomez@igalia.com> Signed-off-by: Samuel Iglesias Gonsalvez <siglesias@igalia.com>

Reference chans from 0 to 4 in the test
Signed-off-by: Miguel Gomez <magomez@igalia.com> Signed-off-by: Samuel Iglesias Gonsalvez <siglesias@igalia.com>
30c73c2d · Miguel Gómez Sexto · Samuel Iglesias Gonsálvez · 583e97ec · 30c73c2d
Commit 30c73c2d authored Oct 17, 2012 by Miguel Gómez Sexto Committed by Samuel Iglesias Gonsálvez Oct 22, 2012
Hide whitespace changes
Inline Side-by-side

Showing with 10 additions and 2 deletions

test-fmctdc.py test/test-fmctdc.py +10 -2

No files found.
--- a/test/test-fmctdc.py
+++ b/test/test-fmctdc.py
@@ -42,6 +42,14 @@ def print_header():
    print ('Website: http://www.ohwr.org/projects/fmc-tdc-sw')
    print ('')

+def chan_mask(chans_str):
+    "returns the binary mask representing a string of channels"
+    mask = 0
+    chans_list = chans_str.split()
+    for i in range(len(chans_list)):
+        mask = mask | (1 << int(chans_list[i]))
+    return mask
+
 class Cli(cmd.Cmd):
    def __init__(self, arg):
        cmd.Cmd.__init__(self)
@@ -197,7 +205,7 @@ class Cli(cmd.Cmd):
            self.libtdc.tdc_get_active_channels(self.tdc, byref(val))
            print val
 	else:
-	    val = c_uint32(int(arg))
+	    val = c_uint32(chan_mask(arg))
            self.libtdc.tdc_set_active_channels(self.tdc, val)

    def do_read (self, arg):
@@ -212,7 +220,7 @@ class Cli(cmd.Cmd):
            print "Invalid arguments"
            return

-        chan = int(args[0])
+        chan = chan_mask(args[0])
        nsamples = int(args[1])

        if (chan < 0) or (chan > 4):