40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
from unittest import skipIf
|
|
|
|
from django.test import SimpleTestCase
|
|
|
|
from ..utils import setup
|
|
|
|
try:
|
|
import numpy
|
|
except ImportError:
|
|
numpy = False
|
|
|
|
|
|
@skipIf(numpy is False, "Numpy must be installed to run these tests.")
|
|
class NumpyTests(SimpleTestCase):
|
|
@setup({"numpy-array-index01": "{{ var.1 }}"})
|
|
def test_numpy_array_index01(self):
|
|
"""
|
|
Numpy's array-index syntax allows a template to access a certain
|
|
item of a subscriptable object.
|
|
"""
|
|
output = self.engine.render_to_string(
|
|
"numpy-array-index01",
|
|
{"var": numpy.array(["first item", "second item"])},
|
|
)
|
|
self.assertEqual(output, "second item")
|
|
|
|
@setup({"numpy-array-index02": "{{ var.5 }}"})
|
|
def test_numpy_array_index02(self):
|
|
"""
|
|
Fail silently when the array index is out of range.
|
|
"""
|
|
output = self.engine.render_to_string(
|
|
"numpy-array-index02",
|
|
{"var": numpy.array(["first item", "second item"])},
|
|
)
|
|
if self.engine.string_if_invalid:
|
|
self.assertEqual(output, "INVALID")
|
|
else:
|
|
self.assertEqual(output, "")
|