From 17be5336b67172afaf070efe5e8e82a551d8e8e7 Mon Sep 17 00:00:00 2001
From: Mark Driver <mdd31@cam.ac.uk>
Date: Wed, 9 Aug 2017 20:20:05 +0100
Subject: [PATCH] added test for the writing.
---
.../test/iotest/polynomialdatawritertest.py | 17 ++++++++++++++++-
.../test/resources/expected_poly_fit2.csv | 3 +++
2 files changed, 19 insertions(+), 1 deletion(-)
create mode 100644 solventmapcreator/test/resources/expected_poly_fit2.csv
diff --git a/solventmapcreator/test/iotest/polynomialdatawritertest.py b/solventmapcreator/test/iotest/polynomialdatawritertest.py
index a94c3a6..1789511 100644
--- a/solventmapcreator/test/iotest/polynomialdatawritertest.py
+++ b/solventmapcreator/test/iotest/polynomialdatawritertest.py
@@ -54,7 +54,22 @@ class PolynomialDataWriterTestCase(unittest.TestCase):
def test_write_poly_data_to_file_split_fit(self):
"""Test to see if expected file is written out.
"""
-
+ expected_file_name = "resources/expected_poly_fit2.csv"
+ actual_file_name = "actual_poly_fit2.csv"
+ poly_file_out = polynomialdatawriter.write_poly_data_to_file_split_fit(self.datapoints,
+ [1], actual_file_name,
+ ['EXAMPLE1',
+ 'EXAMPLE2',
+ 'EXAMPLE3'],
+ ['EXAMPLE1',
+ 'EXAMPLE2',
+ 'EXAMPLE3'])
+ self.assertEqual(0, poly_file_out)
+ with open(expected_file_name, 'r') as exp_file:
+ exp_file_lines = exp_file.readlines()
+ with open(actual_file_name, 'r') as act_file:
+ act_file_lines = act_file.readlines()
+ self.assertListEqual(act_file_lines, exp_file_lines)
def test_write_poly_data_to_file(self):
"""Test to see if expected file is produced.
"""
diff --git a/solventmapcreator/test/resources/expected_poly_fit2.csv b/solventmapcreator/test/resources/expected_poly_fit2.csv
new file mode 100644
index 0000000..bf4273c
--- /dev/null
+++ b/solventmapcreator/test/resources/expected_poly_fit2.csv
@@ -0,0 +1,3 @@
+Order Fit Range RMSE Covariance Coefficients
+1 positive 0.0154303 0.0007143 -0.0500000 1.0714286
+1 negative 0.0154303 0.0007143 -0.0500000 1.0714286
--
GitLab