def test_append(self):
a = pd.DataFrame(['foo'])
data_utils.save_pickle_data('s_pandas.pkl',
a,
append=True,
pandas_format=True,
mode='wb')
np.testing.assert_array_equal(
np.concatenate((self.s.values, a.values)),
data_utils.load_pickle_data('s_pandas.pkl',
pandas_format=True,
mode='rb'))
def test_overwrite(self):
a = pd.DataFrame(['foo'])
data_utils.save_pickle_data('s.pkl',
a,
pandas_format=False,
append=False,
mode='wb')
self.assertEqual(
a.values,
data_utils.load_pickle_data('s.pkl',
pandas_format=False,
mode='rb'))
def test_load_exceptions(self):
with self.assertRaises(exceptions.FileLoadError):
data_utils.load_pickle_data('s2.pkl',
pandas_format=False,
append=False,
mode='rb')
def test_pandas_read(self):
np.testing.assert_array_equal(
self.s.values,
data_utils.load_pickle_data('s_pandas.pkl',
pandas_format=True,
mode='rb').values)
def test_basic_read(self):
self.assertEqual(
self.s.values,
data_utils.load_pickle_data('s.pkl',
pandas_format=False,
mode='rb'))