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