def test_stdout(self): ts = msprime.simulate(10, random_seed=2) parser = cli.stdpopsim_cli_parser() args = parser.parse_args(["AraTha", "2"]) with mock.patch("shutil.copyfileobj") as mocked_copy: cli.write_output(ts, args) mocked_copy.assert_called_once()
def test_stdout(self): ts = msprime.simulate(10, random_seed=2) parser = cli.stdpopsim_cli_parser() args = parser.parse_args(["AraTha", "2"]) with mock.patch("stdpopsim.cli.write_to_stdout") as mocked_func: cli.write_output(ts, args) mocked_func.assert_called_once()
def test_to_file(self): ts = msprime.simulate(10, random_seed=2) parser = cli.stdpopsim_cli_parser() output_file = "mocked.trees" args = parser.parse_args(["HomSap", "2", "-o", output_file]) with mock.patch("tskit.TreeSequence.dump", autospec=True) as mocked_dump: cli.write_output(ts, args) mocked_dump.assert_called_once_with(mock.ANY, output_file)
def test_stdout(self): ts = msprime.simulate(10, random_seed=2) parser = cli.stdpopsim_cli_parser() args = parser.parse_args(["AraTha", "2"]) with mock.patch("shutil.copyfileobj", autospec=True) as mocked_copy: with mock.patch("sys.stdout", autospec=True) as stdout: stdout.buffer = open(os.devnull, "wb") cli.write_output(ts, args) mocked_copy.assert_called_once()