old_contents: # -*- coding: utf-8 -*-
  """
  Test the core grains
  """
  
  from __future__ import absolute_import, unicode_literals
  
  import pytest
  from tests.support.case import ModuleCase
  from tests.support.helpers import slowTest
  
  
  @pytest.mark.windows_whitelisted
  class TestGrainsCore(ModuleCase):
      """
      Test the core grains grains
      """
  
      @slowTest
      def test_grains_passed_to_custom_grain(self):
          """
          test if current grains are passed to grains module functions that have a grains argument
          """
          self.assertEqual(
              self.run_function("grains.get", ["custom_grain_test"]), "itworked"
          )
  
new_contents: """
  Test the core grains
  """
  
  
  import pytest
  from tests.support.case import ModuleCase
  from tests.support.helpers import slowTest
  
  
  @pytest.mark.windows_whitelisted
  class TestGrainsCore(ModuleCase):
      """
      Test the core grains grains
      """
  
      @slowTest
      def test_grains_passed_to_custom_grain(self):
          """
          test if current grains are passed to grains module functions that have a grains argument
          """
          self.assertEqual(
              self.run_function("grains.get", ["custom_grain_test"]), "itworked"
          )
  
subject: Drop Py2 and six on tests/integration/grains/test_custom.py