我想要一个执行cql语句的python脚本并将数据保存到actual_ouput.csv文件中,一旦生成actual_output.csv然后检查expect_result.csv文件(Given)是否其内容相同?< / p>
例如:
expected_result.csv(/src/test/expected_result.csv)
1110003,0,normal,ced74000-af80
2000003,0,critical,ced74000-vd93
4000203,0,normal,ced74000-af91
6004003,0,critical,309ba800-af9a
现在
query = """SELECT * FROM {keyspace}.{table};
""".format(keyspace="mydb", table="Hospital")
query_result = session.execute(query)
with open(actual_output.csv, 'wb') as actua_output_file:
writer = csv.writer(actua_output_file)
writer.writerows([(row.id, row.p_id, row.conditionrow.data) for row in query_result])
上次statemet中的问题始终指定字段名称。但是我们不希望字段名与行相关联(例如:row.id,row.p_id等),因此相同的代码将用于任何表。
一旦数据成功保存到actual_output.csv文件中,然后检查两个文件的内容是否相同。 像这样的东西:
asseertEqual(file1, file2)
建议我使用python脚本??