3f1df8eb2a
Ran pyupgrade on lms/djangoapps/instructor_analytics Ran pyugprade on lms/djangoapps/instructor_task Ran pyupgrade on lms/djangoapps/learner_dashboard
126 lines
4.5 KiB
Python
126 lines
4.5 KiB
Python
""" Tests for analytics.csvs """
|
|
|
|
|
|
import pytest
|
|
from django.test import TestCase
|
|
|
|
from lms.djangoapps.instructor_analytics.csvs import create_csv_response, format_dictlist, format_instances
|
|
|
|
|
|
class TestAnalyticsCSVS(TestCase):
|
|
""" Test analytics rendering of csv files."""
|
|
|
|
def test_create_csv_response_nodata(self):
|
|
header = ['Name', 'Email']
|
|
datarows = []
|
|
|
|
res = create_csv_response('robot.csv', header, datarows)
|
|
assert res['Content-Type'] == 'text/csv'
|
|
assert res['Content-Disposition'] == 'attachment; filename={}'.format('robot.csv')
|
|
assert res.content.strip().decode('utf-8') == '"Name","Email"'
|
|
|
|
def test_create_csv_response(self):
|
|
header = ['Name', 'Email']
|
|
datarows = [['Jim', 'jim@edy.org'], ['Jake', 'jake@edy.org'], ['Jeeves', 'jeeves@edy.org']]
|
|
|
|
res = create_csv_response('robot.csv', header, datarows)
|
|
assert res['Content-Type'] == 'text/csv'
|
|
assert res['Content-Disposition'] == 'attachment; filename={}'.format('robot.csv')
|
|
assert res.content.strip().decode('utf-8') ==\
|
|
'"Name","Email"\r\n"Jim","jim@edy.org"\r\n"Jake","jake@edy.org"\r\n"Jeeves","jeeves@edy.org"'
|
|
|
|
def test_create_csv_response_empty(self):
|
|
header = []
|
|
datarows = []
|
|
|
|
res = create_csv_response('robot.csv', header, datarows)
|
|
assert res['Content-Type'] == 'text/csv'
|
|
assert res['Content-Disposition'] == 'attachment; filename={}'.format('robot.csv')
|
|
assert res.content.strip().decode('utf-8') == ''
|
|
|
|
|
|
class TestAnalyticsFormatDictlist(TestCase):
|
|
""" Test format_dictlist method """
|
|
|
|
def test_format_dictlist(self):
|
|
dictlist = [
|
|
{
|
|
'label1': 'value-1,1',
|
|
'label2': 'value-1,2',
|
|
'label3': 'value-1,3',
|
|
'label4': 'value-1,4',
|
|
},
|
|
{
|
|
'label1': 'value-2,1',
|
|
'label2': 'value-2,2',
|
|
'label3': 'value-2,3',
|
|
'label4': 'value-2,4',
|
|
}
|
|
]
|
|
|
|
features = ['label1', 'label4']
|
|
header, datarows = format_dictlist(dictlist, features)
|
|
|
|
ideal_header = ['label1', 'label4']
|
|
ideal_datarows = [['value-1,1', 'value-1,4'],
|
|
['value-2,1', 'value-2,4']]
|
|
|
|
assert header == ideal_header
|
|
assert datarows == ideal_datarows
|
|
|
|
def test_format_dictlist_empty(self):
|
|
header, datarows = format_dictlist([], [])
|
|
assert header == []
|
|
assert datarows == []
|
|
|
|
def test_create_csv_response(self):
|
|
header = ['Name', 'Email']
|
|
datarows = [['Jim', 'jim@edy.org'], ['Jake', 'jake@edy.org'], ['Jeeves', 'jeeves@edy.org']]
|
|
|
|
res = create_csv_response('robot.csv', header, datarows)
|
|
assert res['Content-Type'] == 'text/csv'
|
|
assert res['Content-Disposition'] == 'attachment; filename={}'.format('robot.csv')
|
|
assert res.content.strip().decode('utf-8') ==\
|
|
'"Name","Email"\r\n"Jim","jim@edy.org"\r\n"Jake","jake@edy.org"\r\n"Jeeves","jeeves@edy.org"'
|
|
|
|
|
|
class TestAnalyticsFormatInstances(TestCase):
|
|
""" test format_instances method """
|
|
|
|
class TestDataClass:
|
|
""" Test class to generate objects for format_instances """
|
|
def __init__(self):
|
|
self.a_var = 'aval'
|
|
self.b_var = 'bval'
|
|
self.c_var = 'cval'
|
|
|
|
@property
|
|
def d_var(self):
|
|
""" accessor to see if they work too """
|
|
return 'dval'
|
|
|
|
def setUp(self):
|
|
super().setUp()
|
|
self.instances = [self.TestDataClass() for _ in range(5)]
|
|
|
|
def test_format_instances_response(self):
|
|
features = ['a_var', 'c_var', 'd_var']
|
|
header, datarows = format_instances(self.instances, features)
|
|
assert header == ['a_var', 'c_var', 'd_var']
|
|
assert datarows == [['aval', 'cval', 'dval'] for _ in range(len(self.instances))]
|
|
|
|
def test_format_instances_response_noinstances(self):
|
|
features = ['a_var']
|
|
header, datarows = format_instances([], features)
|
|
assert header == features
|
|
assert datarows == []
|
|
|
|
def test_format_instances_response_nofeatures(self):
|
|
header, datarows = format_instances(self.instances, [])
|
|
assert header == []
|
|
assert datarows == [[] for _ in range(len(self.instances))]
|
|
|
|
def test_format_instances_response_nonexistantfeature(self):
|
|
with pytest.raises(AttributeError):
|
|
format_instances(self.instances, ['robot_not_a_real_feature'])
|