From 8910a83ffd113c443be65ea5dc4eef1ba45630dc Mon Sep 17 00:00:00 2001 From: aarif Date: Fri, 18 Oct 2019 15:27:08 +0500 Subject: [PATCH] updated the csv writer for python 3 re arranged the imports to fix quality errors --- .../acceptance/tests/discussion/test_cohort_management.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/common/test/acceptance/tests/discussion/test_cohort_management.py b/common/test/acceptance/tests/discussion/test_cohort_management.py index 2a84e90dad..6150d7779a 100644 --- a/common/test/acceptance/tests/discussion/test_cohort_management.py +++ b/common/test/acceptance/tests/discussion/test_cohort_management.py @@ -8,9 +8,12 @@ from __future__ import absolute_import import os import os.path import uuid +import csv +import unicodecsv +import six + from datetime import datetime -import unicodecsv from bok_choy.promise import EmptyPromise from pytz import UTC, utc @@ -322,7 +325,7 @@ class CohortConfigurationTest(EventsTestMixin, UniqueCourseTest, CohortTestMixin """ filename = self.instructor_dashboard_page.get_asset_path(filename) with open(filename, 'w+') as csv_file: - writer = unicodecsv.writer(csv_file) + writer = csv.writer(csv_file) if six.PY3 else unicodecsv.writer(csv_file) for line in csv_text_as_lists: writer.writerow(line) self.addCleanup(os.remove, filename)