def testBidirectionalFalse(self, data, bidirectional):
    paraphrase_problem = paraphrase_ms_coco.ParaphraseGenerationProblem()
    paraphrase_problem.bidirectional = False

    expected_generated_data = [{"inputs": "sentence1", "targets": "sentence2"}]
    actual_generated_data = list(paraphrase_problem
                                 .generate_samples("data_dir",
                                                   "tmp_dir",
                                                   "dataset_split"))
    self.assertEqual(actual_generated_data, expected_generated_data)