예제 #1
0
def test_collection_with_dataframe():
    data = gen_dataframe(default_nb)
    collection, _ = Collection.construct_from_dataframe(name=gen_unique_str(), dataframe=data, primary_field="int64")
    collection.load()
    assert collection.is_empty is False
    assert collection.num_entities == default_nb
    collection.drop()
예제 #2
0
 def test_construct_from_dataframe(self):
     assert type(
         Collection.construct_from_dataframe(
             gen_collection_name(),
             gen_pd_data(default_nb),
             primary_field="int64")[0]) is Collection