def test_dump_to_csv_dumps(self): if os.path.exists( os.path.dirname(os.path.abspath(__file__)) + '/data/output.csv'): os.remove( os.path.dirname(os.path.abspath(__file__)) + '/data/output.csv') field_list = [ 'id_str', 'coordinates.coordinates.0', 'coordinates.coordinates.1', 'user.id_str', 'user.lang', 'lang', 'text', 'user.screen_name', 'user.location', 'user.description', 'created_at', 'user.friends_count', 'user.followers_count', 'retweet_count', 'entities.urls.0.expanded_url', 'entities.urls.1.expanded_url', 'entities.urls.2.expanded_url', 'entities.urls.3.expanded_url', 'entities.urls.4.expanded_url', 'entities.hashtags.0.text', 'entities.hashtags.1.text' ] output_path = os.path.dirname( os.path.realpath(__file__)) + '/' + 'data/output.csv' collection = BsonCollection( os.path.dirname(os.path.realpath(__file__)) + '/' + config['bson']['valid']) collection.dump_to_csv(output_path, field_list) self.assertTrue(os.path.getsize(output_path) > 0) if os.path.exists( os.path.dirname(os.path.abspath(__file__)) + '/data/output.csv'): os.remove( os.path.dirname(os.path.abspath(__file__)) + '/data/output.csv')
def test_dump_to_csv_dumps(self): if os.path.exists(os.path.dirname(os.path.abspath(__file__))+'/data/output.csv'): os.remove(os.path.dirname(os.path.abspath(__file__))+'/data/output.csv') output_path = os.path.dirname(os.path.realpath(__file__)) + '/' + 'data/output.csv' collection = BsonCollection(os.path.dirname(os.path.realpath(__file__)) +'/'+ config['bson']['valid']) collection.dump_to_csv(output_path, ['id_str', 'entities.hashtags.0', 'entities.hashtags.1']) self.assertTrue(os.path.getsize(output_path) > 0) if os.path.exists(os.path.dirname(os.path.abspath(__file__))+'/data/output.csv'): os.remove(os.path.dirname(os.path.abspath(__file__))+'/data/output.csv')