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)