summaryrefslogtreecommitdiff
path: root/tests/template_tests/filter_tests/test_get_digit.py
blob: 4b10ca93edd58d929579b05360047a85fb10cbc6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
from django.template.defaultfilters import get_digit
from django.test import SimpleTestCase


class FunctionTests(SimpleTestCase):
    def test_values(self):
        self.assertEqual(get_digit(123, 1), 3)
        self.assertEqual(get_digit(123, 2), 2)
        self.assertEqual(get_digit(123, 3), 1)
        self.assertEqual(get_digit(123, 4), 0)
        self.assertEqual(get_digit(123, 0), 123)

    def test_string(self):
        self.assertEqual(get_digit("xyz", 0), "xyz")