def panda_factory(self, colnames, rows):
"""
Returns Rows in a Panda DataFrame
:param rows: values selected in Select statement
:param colnames: column names selected
:return: Panda DataFrame
"""
if len(rows) == 0:
return CassandraFrame(session=self.session)
return CassandraFrame(rows, columns=colnames, session=self.session)
def setUp(self):
super(TestQuery, self).setUp()
self.frame = CassandraFrame([["VIN1", "ford", "black", "frank"],
["VIN2", "cyrsler", "blue", "chris"],
["VIN3", "honda", "red", "harry"]],
columns=self.cols,
session=self.session,
table="tester")
self.frame.create_cql_insert()
color text,
owner text,
passengers set<text>,
data blob
);""")
cols = ["id", "car", "color", "owner"]
session = cl.connect("tests")
#df = pd.DataFrame(range(1,5), columns=["a"])
#tmp = CassandraFrame(np.random.randn(10, 2), columns=["id",""], session = session, table="albums")
tmp = CassandraFrame(
[["VIN1", "ford", "black", "frank"], ["VIN2", "cyrsler", "blue", "chris"],
["VIN3", "honda", "red", "harry"]],
columns=cols,
session=session,
table="albums")
tmp.create_cql_insert()
tmp.insert_async()
print "Now see that the data was inserted"
session.execute("""SELECT id, car, color, owner FROM tests.albums""")
print "The description of tests.albumns:"
print cl.keyspaces["tests"].tables["albums"]
print "As opposed to this:"
print cl.metadata.keyspaces["tests"].tables["albums"].export_as_string()
print "Another comparison"
print cl.keyspaces["tests"].tables["sold_cars"]
