from kitten.util import AutoParadigmMixin


class TestAutoParadigmMixin(object):
    def setup_method(self, method):
        self.apm = AutoParadigmMixin()

    def test_first_load(self):
        ret = self.apm.paradigms

        assert 'node' in ret
        assert 'node' in self.apm._paradigms

    def test_second_load(self):
        self.apm._paradigms = {'hehe': True}
        ret = self.apm.paradigms

        assert 'hehe' in ret
        assert 'hehe' in self.apm._paradigms

from kitten.util import AutoParadigmMixin


class TestAutoParadigmMixin(object):
    def setup_method(self, method):
        self.apm = AutoParadigmMixin()

    def test_first_load(self):
        ret = self.apm.paradigms

        assert 'node' in ret
        assert 'node' in self.apm._paradigms

    def test_second_load(self):
        self.apm._paradigms = {'hehe': True}
        ret = self.apm.paradigms

        assert 'hehe' in ret

    def test_override(self):
        self.apm.paradigms = {'hehe': True}

        assert self.apm._paradigms == {'hehe': True}

Add tests for APM overrides
Add tests for APM overrides

  from kitten.util import AutoParadigmMixin
  
  
  class TestAutoParadigmMixin(object):
      def setup_method(self, method):
          self.apm = AutoParadigmMixin()
  
      def test_first_load(self):
          ret = self.apm.paradigms
  
          assert 'node' in ret
          assert 'node' in self.apm._paradigms
  
      def test_second_load(self):
          self.apm._paradigms = {'hehe': True}
          ret = self.apm.paradigms
  
          assert 'hehe' in ret
-         assert 'hehe' in self.apm._paradigms
  
+     def test_override(self):
+         self.apm.paradigms = {'hehe': True}
+ 
+         assert self.apm._paradigms == {'hehe': True}
+ 
Add tests for APM overrides
## Code Before:
from kitten.util import AutoParadigmMixin


class TestAutoParadigmMixin(object):
    def setup_method(self, method):
        self.apm = AutoParadigmMixin()

    def test_first_load(self):
        ret = self.apm.paradigms

        assert 'node' in ret
        assert 'node' in self.apm._paradigms

    def test_second_load(self):
        self.apm._paradigms = {'hehe': True}
        ret = self.apm.paradigms

        assert 'hehe' in ret
        assert 'hehe' in self.apm._paradigms

## Instruction:
Add tests for APM overrides
## Code After:
from kitten.util import AutoParadigmMixin


class TestAutoParadigmMixin(object):
    def setup_method(self, method):
        self.apm = AutoParadigmMixin()

    def test_first_load(self):
        ret = self.apm.paradigms

        assert 'node' in ret
        assert 'node' in self.apm._paradigms

    def test_second_load(self):
        self.apm._paradigms = {'hehe': True}
        ret = self.apm.paradigms

        assert 'hehe' in ret

    def test_override(self):
        self.apm.paradigms = {'hehe': True}

        assert self.apm._paradigms == {'hehe': True}
