def test_put_df_success_dataframe_to_csv(self):
o = pandas.DataFrame.from_dict(self.data)
key = MY_PREFIX + '/key1.csv'
put_df(self.client, o, MY_BUCKET, key, format='csv')
body = pandas.read_csv(
self.client.get_object(Bucket=MY_BUCKET, Key=key)['Body'])
self.assertSequenceEqual(list(o.columns), list(body.columns))
self.assertSequenceEqual(o.iloc[0].tolist(), body.iloc[0].tolist())
def test_put_df_success_dataframe_to_multiple_csv(self):
o = pandas.DataFrame.from_dict(self.data)
key = MY_PREFIX + '/key1.csv'
obj_1_key = MY_PREFIX + '/key1/key1.1.csv'
obj_2_key = MY_PREFIX + '/key1/key1.2.csv'
put_df(self.client, o, MY_BUCKET, key, format='csv', sep=';', parts=2)
body_1 = pandas.read_csv(self.client.get_object(Bucket=MY_BUCKET,
Key=obj_1_key)['Body'],
sep=';')
body_2 = pandas.read_csv(self.client.get_object(Bucket=MY_BUCKET,
Key=obj_2_key)['Body'],
sep=';')
self.assertTrue(
o.equals(pandas.concat([body_1, body_2]).reset_index(drop=True)))
def test_put_df_failure_dataframe_to_unknown_format(self):
o = pandas.DataFrame.from_dict(self.data)
key = MY_PREFIX + '/key1.txt'
with self.assertRaises(AssertionError):
put_df(self.client, o, MY_BUCKET, key, format='txt')
def test_put_df_failure_unknown_type(self):
o = 'awesome body'
key = MY_PREFIX + '/key1'
with self.assertRaises(TypeError):
put_df(self.client, o, MY_BUCKET, key)
