def test_get_commonsenseqa_twice(self): get_commonsenseqa() with mock.patch("lineflow.datasets.commonsenseqa.pickle", autospec=True) as mock_pickle: get_commonsenseqa() mock_pickle.dump.assert_not_called() self.assertEqual(mock_pickle.load.call_count, 1)
def test_get_commonsenseqa(self): raw = get_commonsenseqa() self.assertIn("train", raw) self.assertIn("dev", raw) self.assertIn("test", raw)