From 3c3bb34b1b12fdcfeb84bc34fd806d06b8b13d5b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andr=C3=A1s=20Veres-Szentkir=C3=A1lyi?= <vsza@vsza.hu>
Date: Mon, 22 Feb 2016 17:35:25 +0100
Subject: [PATCH] codegen: added simple test suite

---
 pysstv/examples/codegen.py | 27 ++++++++++++++++++++++++++-
 1 file changed, 26 insertions(+), 1 deletion(-)

diff --git a/pysstv/examples/codegen.py b/pysstv/examples/codegen.py
index 817e1e3..68fb9cd 100644
--- a/pysstv/examples/codegen.py
+++ b/pysstv/examples/codegen.py
@@ -109,6 +109,31 @@ def gen_matches(same_as, history, n):
                     cur_start = i
                     cur_end = tmp
 
+def test():
+    from subprocess import Popen, PIPE, check_output
+    from os import remove, path
+    from PIL import Image
+    import struct
+    exe = './codegen-test-executable'
+    try:
+        for sstv_class in supported:
+            print 'Testing', sstv_class
+            gcc = Popen(['gcc', '-xc', '-o', exe, '-'], stdin=PIPE)
+            with open(path.join(path.dirname(__file__), 'codeman.c')) as cm:
+                gcc.communicate(cm.read().replace('#include "codegen.c"', '\n'.join(main(sstv_class))))
+            gen = check_output([exe])
+            img = Image.open("320x256rgb.png")
+            sstv = sstv_class(img, 44100, 16)
+            for n, (freq, msec) in enumerate(sstv.gen_freq_bits()):
+                assert gen[n * 8:(n + 1) * 8] == struct.pack('ff', freq, msec)
+        print 'OK'
+    finally:
+        remove(exe)
+
 
 if __name__ == '__main__':
-    main()
+    from sys import argv
+    if len(argv) > 1 and argv[1] == 'test':
+        test()
+    else:
+        print '\n'.join(main())