1 files changed, 26 insertions, 0 deletions
diff --git a/tests.py b/tests.py
new file mode 100644
index 0000000..4e9b8ee
--- /dev/null
+++ b/tests.py
@@ -0,0 +1,26 @@
+import filecmp, itertools, os, sys, tempfile
+
+pybcddpath = os.path.realpath(os.path.join(os.path.dirname(__file__), 'pybcdd'))
+sys.path.append(pybcddpath)
+
+from dealconvert import DealConverter
+
+import pytest
+
+
+formats = ['ber', 'bhg', 'bri', 'cds', 'csv', 'dge', 'dlm', 'dup', 'lin', 'pbn', 'rzd']
+format_files = ['test.' + f for f in formats] + ['test-jfr.pbn']
+test_data = [(d[0], d[1], d[1] == 'test-jfr.pbn') for d in itertools.product(format_files, format_files)]
+
+
+@pytest.mark.parametrize("input_file, output_file, jfr_only", test_data)
+def test_conversion(input_file, output_file, jfr_only):
+    input_path = os.path.join(os.path.dirname(__file__), 'test', input_file)
+    output_path = os.path.join(tempfile.mkdtemp(), output_file)
+    ref_path = os.path.join(os.path.dirname(__file__), 'test', output_file)
+    print(input_path, output_path, ref_path)
+    converter = DealConverter(
+        input_path,
+        jfr_only=jfr_only)
+    converter.output([output_path])
+    assert filecmp.cmp(output_path, ref_path)