old_contents: """
  Tests for `kolibri` module.
  """
  from __future__ import absolute_import, print_function, unicode_literals
  
  import kolibri
  
  from .base import KolibriTestBase
  
  
  class TestKolibriVersion(KolibriTestBase):
  
      def test_version(self):
          """
          Test that the major version is set as expected
          """
          major_version = ".".join(map(str, kolibri.VERSION[:2]))
          self.assertIn(major_version, kolibri.__version__)
  
new_contents: """
  Tests for `kolibri` module.
  """
  from __future__ import absolute_import, print_function, unicode_literals
  
  import kolibri
  from kolibri.utils.version import get_version
  
  from .base import KolibriTestBase
  
  
  class TestKolibriVersion(KolibriTestBase):
  
      def test_version(self):
          """
          Test that the major version is set as expected
          """
          version_instance = get_version()
          self.assertIn(version_instance.major_version, kolibri.__version__)
  
subject: Update version testing to use Version class.