"""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))