def testDataFrame(self):
df = DataFrame(self.table)
self.assertEqual(3, df.count().execute())
self.assertEqual(1, df[df.name == 'name1'].count().execute())
res = df[df.name.contains('中文')].execute()
self.assertGreaterEqual(len(res), 0)
def testCreateDataFrameFromPartition(self):
from odps.types import PartitionSpec
test_table_name = tn('pyodps_test_dataframe_partition')
schema = Schema.from_lists(['id', 'name'], ['bigint', 'string'], ['ds'], ['string'])
self.odps.delete_table(test_table_name, if_exists=True)
table = self.odps.create_table(test_table_name, schema)
with table.open_writer('ds=today', create_partition=True) as w:
w.write([[1, 'name1'], [2, 'name2'], [3, 'name3']])
try:
df = DataFrame(table.get_partition('ds=today'))
self.assertEqual(df.count().execute(), 3)
df = table.get_partition('ds=today').to_df()
partition = df.data
self.assertIs(partition.table, table)
self.assertEqual(partition.partition_spec, PartitionSpec('ds=today'))
self.assertEqual(df.count().execute(), 3)
finally:
table.drop()
def testDataFrame(self):
df = DataFrame(self.table)
self.assertEqual(3, df.count().execute())
self.assertEqual(1, df[df.name == 'name1'].count())
def testDataFrame(self):
df = DataFrame(self.table)
self.assertEqual(3, df.count().execute())
self.assertEqual(1, df[df.name == 'name1'].count())