
import unittest

from columns.templatetags.columns import rows, columns


class TestColumns(unittest.TestCase):

    def test_columns(self):
        data = range(7)
        result = rows(data, 2)
        expected = [[0, 1, 2, 3], [4, 5, 6]]
        self.assertEqual(result, expected)


import unittest

from columns.templatetags.columns import columns


class TestColumns(unittest.TestCase):

    def test_columns(self):
        data = range(7)
        result = columns(data, 2)
        expected = [[0, 1, 2, 3], [4, 5, 6]]
        self.assertEqual(result, expected)

Fix tests to match updated defs.
Fix tests to match updated defs.

  
  import unittest
  
- from columns.templatetags.columns import rows, columns
+ from columns.templatetags.columns import columns
  
  
  class TestColumns(unittest.TestCase):
  
      def test_columns(self):
          data = range(7)
-         result = rows(data, 2)
+         result = columns(data, 2)
          expected = [[0, 1, 2, 3], [4, 5, 6]]
          self.assertEqual(result, expected)
  
Fix tests to match updated defs.
## Code Before:

import unittest

from columns.templatetags.columns import rows, columns


class TestColumns(unittest.TestCase):

    def test_columns(self):
        data = range(7)
        result = rows(data, 2)
        expected = [[0, 1, 2, 3], [4, 5, 6]]
        self.assertEqual(result, expected)

## Instruction:
Fix tests to match updated defs.
## Code After:

import unittest

from columns.templatetags.columns import columns


class TestColumns(unittest.TestCase):

    def test_columns(self):
        data = range(7)
        result = columns(data, 2)
        expected = [[0, 1, 2, 3], [4, 5, 6]]
        self.assertEqual(result, expected)
