import unittest

from kafka_influxdb.encoder import echo_encoder

class TestEchoEncoder(unittest.TestCase):

    def setUp(self):
        self.encoder = echo_encoder.Encoder()
        self.messages = [
            "yeaal",
            ["this", "is", "a", "list"],
            {'hash': {'maps': 'rule'}},
            42,
            42.23
        ]

    def test_encode(self):
        for msg in self.messages:
            yield check_encode, msg

    def check_encode(self, msg):
        """ Output must be same as input for echo sender """
        self.assertEqual(self.encoder.encode(msg), msg)

import unittest
from kafka_influxdb.encoder import echo_encoder


class TestEchoEncoder(unittest.TestCase):

    def setUp(self):
        self.encoder = echo_encoder.Encoder()
        self.messages = [
            "yeaal",
            ["this", "is", "a", "list"],
            {'hash': {'maps': 'rule'}},
            42,
            42.23
        ]

    def test_encode(self):
        for msg in self.messages:
            yield self.check_encode(msg)

    def check_encode(self, msg):
        """ Output must be same as input for echo sender """
        self.assertEqual(self.encoder.encode(msg), msg)

Fix unit test for echo encoder
Fix unit test for echo encoder

  import unittest
+ from kafka_influxdb.encoder import echo_encoder
  
- from kafka_influxdb.encoder import echo_encoder
  
  class TestEchoEncoder(unittest.TestCase):
  
      def setUp(self):
          self.encoder = echo_encoder.Encoder()
          self.messages = [
              "yeaal",
              ["this", "is", "a", "list"],
              {'hash': {'maps': 'rule'}},
              42,
              42.23
          ]
  
      def test_encode(self):
          for msg in self.messages:
-             yield check_encode, msg
+             yield self.check_encode(msg)
  
      def check_encode(self, msg):
          """ Output must be same as input for echo sender """
          self.assertEqual(self.encoder.encode(msg), msg)
  
Fix unit test for echo encoder
## Code Before:
import unittest

from kafka_influxdb.encoder import echo_encoder

class TestEchoEncoder(unittest.TestCase):

    def setUp(self):
        self.encoder = echo_encoder.Encoder()
        self.messages = [
            "yeaal",
            ["this", "is", "a", "list"],
            {'hash': {'maps': 'rule'}},
            42,
            42.23
        ]

    def test_encode(self):
        for msg in self.messages:
            yield check_encode, msg

    def check_encode(self, msg):
        """ Output must be same as input for echo sender """
        self.assertEqual(self.encoder.encode(msg), msg)

## Instruction:
Fix unit test for echo encoder
## Code After:
import unittest
from kafka_influxdb.encoder import echo_encoder


class TestEchoEncoder(unittest.TestCase):

    def setUp(self):
        self.encoder = echo_encoder.Encoder()
        self.messages = [
            "yeaal",
            ["this", "is", "a", "list"],
            {'hash': {'maps': 'rule'}},
            42,
            42.23
        ]

    def test_encode(self):
        for msg in self.messages:
            yield self.check_encode(msg)

    def check_encode(self, msg):
        """ Output must be same as input for echo sender """
        self.assertEqual(self.encoder.encode(msg), msg)
