82 lines
2.4 KiB
Python
82 lines
2.4 KiB
Python
# lint-amnesty, pylint: disable=missing-module-docstring
|
|
|
|
import unittest
|
|
import xml.sax.saxutils as saxutils
|
|
|
|
from lxml import etree
|
|
|
|
from xmodule.capa import customrender
|
|
from xmodule.capa.tests.helpers import mock_capa_system
|
|
|
|
# just a handy shortcut
|
|
lookup_tag = customrender.registry.get_class_for_tag
|
|
|
|
|
|
def extract_context(xml):
|
|
"""
|
|
Given an xml element corresponding to the output of test_capa_system.render_template, get back the
|
|
original context
|
|
"""
|
|
return eval(xml.text) # lint-amnesty, pylint: disable=eval-used
|
|
|
|
|
|
def quote_attr(s):
|
|
return saxutils.quoteattr(s)[1:-1] # don't want the outer quotes
|
|
|
|
|
|
class HelperTest(unittest.TestCase):
|
|
'''
|
|
Make sure that our helper function works!
|
|
'''
|
|
|
|
def check(self, d):
|
|
xml = etree.XML(mock_capa_system().render_template('blah', d))
|
|
assert d == extract_context(xml)
|
|
|
|
def test_extract_context(self):
|
|
self.check({})
|
|
self.check({1, 2})
|
|
self.check({'id', 'an id'})
|
|
self.check({'with"quote', 'also"quote'})
|
|
|
|
|
|
class SolutionRenderTest(unittest.TestCase):
|
|
'''
|
|
Make sure solutions render properly.
|
|
'''
|
|
|
|
def test_rendering(self):
|
|
solution = 'To compute unicorns, count them.'
|
|
xml_str = """<solution id="solution_12">{s}</solution>""".format(s=solution)
|
|
element = etree.fromstring(xml_str)
|
|
|
|
renderer = lookup_tag('solution')(mock_capa_system(), element)
|
|
|
|
assert renderer.id == 'solution_12'
|
|
|
|
# Our test_capa_system "renders" templates to a div with the repr of the context.
|
|
xml = renderer.get_html()
|
|
context = extract_context(xml)
|
|
assert context == {'id': 'solution_12'}
|
|
|
|
|
|
class MathRenderTest(unittest.TestCase):
|
|
'''
|
|
Make sure math renders properly.
|
|
'''
|
|
|
|
def check_parse(self, latex_in, mathjax_out): # lint-amnesty, pylint: disable=missing-function-docstring
|
|
xml_str = """<math>{tex}</math>""".format(tex=latex_in)
|
|
element = etree.fromstring(xml_str)
|
|
|
|
renderer = lookup_tag('math')(mock_capa_system(), element)
|
|
|
|
assert renderer.mathstr == mathjax_out
|
|
|
|
def test_parsing(self):
|
|
self.check_parse('$abc$', '[mathjaxinline]abc[/mathjaxinline]')
|
|
self.check_parse('$abc', '$abc')
|
|
self.check_parse(r'$\displaystyle 2+2$', '[mathjax] 2+2[/mathjax]')
|
|
|
|
# NOTE: not testing get_html yet because I don't understand why it's doing what it's doing.
|