diff options
Diffstat (limited to 'tests/unit/computations')
| -rw-r--r-- | tests/unit/computations/test_biweight.py | 21 | ||||
| -rw-r--r-- | tests/unit/computations/test_correlation.py | 6 |
2 files changed, 25 insertions, 2 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)) diff --git a/tests/unit/computations/test_correlation.py b/tests/unit/computations/test_correlation.py index 5746adf..606f96d 100644 --- a/tests/unit/computations/test_correlation.py +++ b/tests/unit/computations/test_correlation.py @@ -98,12 +98,14 @@ class TestCorrelation(TestCase): self.assertEqual(results, expected_results) - def test_bicor(self): + @mock.patch("gn3.computations.correlations.call_biweight_script") + def test_bicor(self, mock_biweight): """Test for doing biweight mid correlation """ + mock_biweight.return_value = (1.0, 0.0) results = do_bicor(x_val=[1, 2, 3], y_val=[4, 5, 6]) - self.assertEqual(results, (0.0, 0.0) + self.assertEqual(results, (1.0, 0.0) ) @mock.patch("gn3.computations.correlations.compute_corr_coeff_p_value") |