First 3 tests added for fsm class

git-svn-id: http://gnuradio.org/svn/gnuradio/trunk@3229 221aa14e-8319-0410-a670-987f0aec2ac5

diff --git a/gr-trellis/src/python/awgn1o2_4.fsm b/gr-trellis/src/python/awgn1o2_4.fsm
new file mode 100644 (file)
index 0000000..9245c69
--- /dev/null
+++ b/gr-trellis/src/python/awgn1o2_4.fsm
@@ -0,0 +1,15 @@
+2 4 4
+
+0 2
+0 2
+1 3
+1 3
+
+0 3
+3 0
+1 2
+2 1
+
+AWGN CC from Proakis-Salehi pg 779
+
+

diff --git a/gr-trellis/src/python/qa_trellis.py b/gr-trellis/src/python/qa_trellis.py
index c47ee2881d3dd7d5e9a552b140c16c3733f64a73..e4866e7f924a9dd3f30f00a0b56b7a550a2785a1 100755 (executable)
--- a/gr-trellis/src/python/qa_trellis.py
+++ b/gr-trellis/src/python/qa_trellis.py
@@ -31,5 +31,32 @@ class qa_trellis (gr_unittest.TestCase):
     def tearDown (self):
         self.fg = None
 
+    def test_001_fsm (self):
+        I = 2
+        S = 4
+        O = 4
+        NS = (0, 2, 0, 2, 1, 3, 1, 3)
+        OS = (0, 3, 3, 0, 1, 2, 2, 1)
+        f = trellis.fsm(I,S,O,NS,OS)
+        self.assertEqual((I,S,O,NS,OS),(f.I(),f.S(),f.O(),f.NS(),f.OS()))
+
+    def test_002_fsm (self):
+        I = 2
+        S = 4
+        O = 4
+        NS = (0, 2, 0, 2, 1, 3, 1, 3)
+        OS = (0, 3, 3, 0, 1, 2, 2, 1)
+        f = trellis.fsm(I,S,O,NS,OS)
+        g = trellis.fsm(f)
+        self.assertEqual((g.I(),g.S(),g.O(),g.NS(),g.OS()),(f.I(),f.S(),f.O(),f.NS(),f.OS()))
+
+    def test_003_fsm (self):
+        I = 2
+        S = 4
+        O = 4
+        NS = (0, 2, 0, 2, 1, 3, 1, 3)
+        OS = (0, 3, 3, 0, 1, 2, 2, 1)
+        f = trellis.fsm("awgn1o2_4.fsm")
+        self.assertEqual((I,S,O,NS,OS),(f.I(),f.S(),f.O(),f.NS(),f.OS()))
 if __name__ == '__main__':
     gr_unittest.main ()