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()
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