Commit 30be9af3 by Brian Wilson

Disable pylint E1103 in configuration.

Change-Id: Id11f469b3fd8372c6b85280e9a8882595e62b76b
parent 68abaf6e
......@@ -23,7 +23,7 @@ class CourseEnrollmentCountMixin(object):
# Not assuming any encoding, course_id will be read as plain string
data = read_tsv(input_file, names)
data.date = pandas.to_datetime(data.date) # pylint: disable=maybe-no-member
data.date = pandas.to_datetime(data.date)
return data
def initialize_daily_count(self, course_date_count_data):
......@@ -75,7 +75,7 @@ class CourseEnrollmentCountMixin(object):
# Flag values before the offset day with NaN,
# since they are not "available".
not_available = count_by_day.index < date
count_by_day.loc[not_available, course_id] = numpy.NaN # pylint: disable=maybe-no-member
count_by_day.loc[not_available, course_id] = numpy.NaN
def calculate_total_enrollment(self, count_by_day, offsets=None):
"""
......@@ -174,7 +174,7 @@ class EnrollmentsByWeek(luigi.Task, CourseEnrollmentCountMixin):
statuses = self.read_statuses()
with self.output().open('w') as output_file: # pylint: disable=maybe-no-member
with self.output().open('w') as output_file:
self.save_output(weekly_enrollment_totals, statuses, output_file)
def read_source(self):
......
......@@ -97,7 +97,7 @@ class WeeklyIncrementalUsersAndEnrollments(luigi.Task, AllCourseEnrollmentCountM
weekly_registration_changes, weekly_overall_enrollment
)
with self.output().open('w') as output_file: # pylint: disable=maybe-no-member
with self.output().open('w') as output_file:
self.save_output(weekly_report, output_file)
def read_enrollments(self):
......
......@@ -14,7 +14,6 @@ from edx.analytics.tasks.reports.incremental_enrollments import WeeklyIncrementa
class TestWeeklyIncrementalUsersAndEnrollments(unittest.TestCase):
"""Tests for WeeklyIncrementalUsersAndEnrollments class."""
# pylint: disable=maybe-no-member
@staticmethod
def row_label(row_name):
......
......@@ -111,8 +111,8 @@ class TestWeeklyAllUsersAndEnrollments(unittest.TestCase):
"""
res = self.run_task(source, '2013-01-21', 4)
weeks = set(['2012-12-31', '2013-01-07', '2013-01-14', '2013-01-21'])
self.assertEqual(weeks, set(str(w) for w in res.columns)) # pylint: disable=maybe-no-member
total_enrollment = res.loc[TOTAL_ENROLLMENT_ROWNAME] # pylint: disable=maybe-no-member
self.assertEqual(weeks, set(str(w) for w in res.columns))
total_enrollment = res.loc[TOTAL_ENROLLMENT_ROWNAME]
self.assertTrue(isnan(total_enrollment['2012-12-31'])) # no data
self.assertEqual(total_enrollment['2013-01-07'], 10)
self.assertEqual(total_enrollment['2013-01-14'], 20)
......@@ -131,7 +131,7 @@ class TestWeeklyAllUsersAndEnrollments(unittest.TestCase):
course_2 2013-02-15 -2
"""
res = self.run_task(source, '2013-02-18', 2)
total_enrollment = res.loc[TOTAL_ENROLLMENT_ROWNAME] # pylint: disable=maybe-no-member
total_enrollment = res.loc[TOTAL_ENROLLMENT_ROWNAME]
self.assertEqual(total_enrollment['2013-02-11'], 13)
self.assertEqual(total_enrollment['2013-02-18'], 24)
......@@ -141,7 +141,7 @@ class TestWeeklyAllUsersAndEnrollments(unittest.TestCase):
course_3 2013-03-15 6
"""
res = self.run_task(None, '2013-03-28', 6, offset=offset)
total_enrollment = res.loc[TOTAL_ENROLLMENT_ROWNAME] # pylint: disable=maybe-no-member
total_enrollment = res.loc[TOTAL_ENROLLMENT_ROWNAME]
self.assertTrue(isnan(total_enrollment['2013-02-21'])) # no data
self.assertTrue(isnan(total_enrollment['2013-02-28'])) # no data
self.assertEqual(total_enrollment['2013-03-07'], 10)
......@@ -161,7 +161,7 @@ class TestWeeklyAllUsersAndEnrollments(unittest.TestCase):
2013-02-28 10
"""
res = self.run_task(None, '2013-03-28', 6, offset=offset, history=history)
total_enrollment = res.loc[TOTAL_ENROLLMENT_ROWNAME] # pylint: disable=maybe-no-member
total_enrollment = res.loc[TOTAL_ENROLLMENT_ROWNAME]
self.assertEqual(total_enrollment['2013-02-21'], 4)
self.assertEqual(total_enrollment['2013-02-28'], 10)
self.assertEqual(total_enrollment['2013-03-07'], 10)
......@@ -181,7 +181,7 @@ class TestWeeklyAllUsersAndEnrollments(unittest.TestCase):
2013-03-21 22
"""
res = self.run_task(None, '2013-03-28', 6, offset=offset, history=history)
total_enrollment = res.loc[TOTAL_ENROLLMENT_ROWNAME] # pylint: disable=maybe-no-member
total_enrollment = res.loc[TOTAL_ENROLLMENT_ROWNAME]
print total_enrollment
self.assertEqual(total_enrollment['2013-02-21'], 5)
self.assertEqual(total_enrollment['2013-02-28'], 9)
......@@ -200,7 +200,7 @@ class TestWeeklyAllUsersAndEnrollments(unittest.TestCase):
res = self.run_task(source.encode('utf-8'), '2013-04-02', 1)
self.assertEqual(res.loc[TOTAL_ENROLLMENT_ROWNAME]['2013-04-02'], 2) # pylint: disable=maybe-no-member
self.assertEqual(res.loc[TOTAL_ENROLLMENT_ROWNAME]['2013-04-02'], 2)
def test_task_urls(self):
date = datetime.date(2013, 01, 20)
......
......@@ -30,8 +30,8 @@ class AllCourseEnrollmentCountMixin(CourseEnrollmentCountMixin):
names = ['date', 'count']
data = read_tsv(input_file, names)
data.date = pandas.to_datetime(data.date) # pylint: disable=maybe-no-member
data = data.set_index('date') # pylint: disable=maybe-no-member
data.date = pandas.to_datetime(data.date)
data = data.set_index('date')
date_range = pandas.date_range(min(data.index), max(data.index))
data = data.reindex(date_range)
......@@ -192,7 +192,7 @@ class WeeklyAllUsersAndEnrollments(luigi.Task, AllCourseEnrollmentCountMixin):
self.weeks,
)
with self.output().open('w') as output_file: # pylint: disable=maybe-no-member
with self.output().open('w') as output_file:
self.save_output(total_counts_by_week, output_file)
def read_source(self):
......@@ -222,7 +222,7 @@ class WeeklyAllUsersAndEnrollments(luigi.Task, AllCourseEnrollmentCountMixin):
"""
data = None
if self.input().get('offsets'):
with self.input()['offsets'].open('r') as offset_file: # pylint: disable=maybe-no-member
with self.input()['offsets'].open('r') as offset_file:
data = self.read_course_date_count_tsv(offset_file)
return data
......@@ -239,7 +239,7 @@ class WeeklyAllUsersAndEnrollments(luigi.Task, AllCourseEnrollmentCountMixin):
"""
data = None
if self.input().get('history'):
with self.input()['history'].open('r') as history_file: # pylint: disable=maybe-no-member
with self.input()['history'].open('r') as history_file:
data = self.read_total_count_tsv(history_file)
return data
......
......@@ -57,7 +57,7 @@ disable=
# R0912: Too many branches
# R0913: Too many arguments
# R0914: Too many local variables
C0302,R0201,R0901,R0902,R0903,R0904,R0911,R0912,R0913,R0914
C0302,R0201,R0901,R0902,R0903,R0904,R0911,R0912,R0913,R0914,E1103
[REPORTS]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment