def test_needs_p(self):
fileroot = path.join(self.tempdir, 'this_temp.{}')
temppath = fileroot.format('dta')
self.df.to_stata(temppath)
try_pickle(temppath)
result_list = glob.glob(path.join(self.tempdir, '*'))
expect_list = [fileroot.format(x) for x in ('pkl', 'dta')]
assert set(result_list) == set(expect_list)
def test_already_p(self):
temppath = path.join(self.tempdir, 'this_temp.pkl')
self.df.to_pickle(temppath)
result = try_pickle(temppath)
assert_frame_equal(self.df, result)
file_list = glob.glob(path.join(self.tempdir, '*'))
assert len(file_list) == 1