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..3045f78
--- /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 call_biweight_script
+
+
+class TestBiweight(TestCase):
+    """test class for biweight"""
+
+    @mock.patch("gn3.computations.biweight.subprocess.check_output")
+    def test_call_biweight_script(self, mock_check_output):
+        """test for call_biweight_script func"""
+        mock_check_output.return_value = "0.1 0.5"
+        results = call_biweight_script(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))