from __future__ import unicode_literals
from tests.support.case import ModuleCase


class PillarIncludeTest(ModuleCase):

    def test_pillar_include(self):
        '''
        Test pillar include
        '''
        ret = self.minion_run('pillar.items')
        assert 'a' in ret['element']
        assert ret['element']['a'] == {'a': ['Entry A']}
        assert 'b' in ret['element']
        assert ret['element']['b'] == {'b': ['Entry B']}

    def test_pillar_glob_include(self):
        '''
        Test pillar include via glob pattern
        '''
        ret = self.minion_run('pillar.items')
        assert 'glob-a' in ret
        assert ret['glob-a'] == 'Entry A'
        assert 'glob-b' in ret
        assert ret['glob-b'] == 'Entry B'

'''
Pillar include tests
'''
from __future__ import unicode_literals

from tests.support.case import ModuleCase


class PillarIncludeTest(ModuleCase):

    def test_pillar_include(self):
        '''
        Test pillar include
        '''
        ret = self.minion_run('pillar.items')
        assert 'a' in ret['element']
        assert ret['element']['a'] == {'a': ['Entry A']}
        assert 'b' in ret['element']
        assert ret['element']['b'] == {'b': ['Entry B']}

    def test_pillar_glob_include(self):
        '''
        Test pillar include via glob pattern
        '''
        ret = self.minion_run('pillar.items')
        assert 'glob-a' in ret
        assert ret['glob-a'] == 'Entry A'
        assert 'glob-b' in ret
        assert ret['glob-b'] == 'Entry B'

Use file encoding and add docstring
Use file encoding and add docstring

+ '''
+ Pillar include tests
+ '''
  from __future__ import unicode_literals
+ 
  from tests.support.case import ModuleCase
  
  
  class PillarIncludeTest(ModuleCase):
  
      def test_pillar_include(self):
          '''
          Test pillar include
          '''
          ret = self.minion_run('pillar.items')
          assert 'a' in ret['element']
          assert ret['element']['a'] == {'a': ['Entry A']}
          assert 'b' in ret['element']
          assert ret['element']['b'] == {'b': ['Entry B']}
  
      def test_pillar_glob_include(self):
          '''
          Test pillar include via glob pattern
          '''
          ret = self.minion_run('pillar.items')
          assert 'glob-a' in ret
          assert ret['glob-a'] == 'Entry A'
          assert 'glob-b' in ret
          assert ret['glob-b'] == 'Entry B'
  
Use file encoding and add docstring
## Code Before:
from __future__ import unicode_literals
from tests.support.case import ModuleCase


class PillarIncludeTest(ModuleCase):

    def test_pillar_include(self):
        '''
        Test pillar include
        '''
        ret = self.minion_run('pillar.items')
        assert 'a' in ret['element']
        assert ret['element']['a'] == {'a': ['Entry A']}
        assert 'b' in ret['element']
        assert ret['element']['b'] == {'b': ['Entry B']}

    def test_pillar_glob_include(self):
        '''
        Test pillar include via glob pattern
        '''
        ret = self.minion_run('pillar.items')
        assert 'glob-a' in ret
        assert ret['glob-a'] == 'Entry A'
        assert 'glob-b' in ret
        assert ret['glob-b'] == 'Entry B'

## Instruction:
Use file encoding and add docstring
## Code After:
'''
Pillar include tests
'''
from __future__ import unicode_literals

from tests.support.case import ModuleCase


class PillarIncludeTest(ModuleCase):

    def test_pillar_include(self):
        '''
        Test pillar include
        '''
        ret = self.minion_run('pillar.items')
        assert 'a' in ret['element']
        assert ret['element']['a'] == {'a': ['Entry A']}
        assert 'b' in ret['element']
        assert ret['element']['b'] == {'b': ['Entry B']}

    def test_pillar_glob_include(self):
        '''
        Test pillar include via glob pattern
        '''
        ret = self.minion_run('pillar.items')
        assert 'glob-a' in ret
        assert ret['glob-a'] == 'Entry A'
        assert 'glob-b' in ret
        assert ret['glob-b'] == 'Entry B'
