Merge pull request #20 from genenetwork/feature/biweight-correlation

add biweight r script and tests

1 files changed, 21 insertions, 0 deletions

diff --git a/tests/unit/computations/test_biweight.py b/tests/unit/computations/test_biweight.py
new file mode 100644
index 0000000..ad404f1
--- /dev/null
+++ b/tests/unit/computations/test_biweight.py
@@ -0,0 +1,21 @@
+"""test for biweight script"""
+from unittest import TestCase
+from unittest import mock
+
+from gn3.computations.biweight import calculate_biweight_corr
+
+
+class TestBiweight(TestCase):
+    """test class for biweight"""
+
+    @mock.patch("gn3.computations.biweight.subprocess.check_output")
+    def test_calculate_biweight_corr(self, mock_check_output):
+        """test for calculate_biweight_corr func"""
+        mock_check_output.return_value = "0.1 0.5"
+        results = calculate_biweight_corr(command="Rscript",
+                                          path_to_script="./r_script.R",
+                                          trait_vals=[
+                                              1.2, 1.1, 1.9],
+                                          target_vals=[1.9, 0.4, 1.1])
+
+        self.assertEqual(results, (0.1, 0.5))