import unittest

from address import has_reflection, is_compatible


class TestAddress(unittest.TestCase):
    def test_has_reflection(self):
        assert has_reflection(['mnop']) == False
        assert has_reflection(['abba', 'qrst']) == True

    def test_is_compatible(self):
        assert is_compatible('abba[mnop]qrst') == True
        assert is_compatible('abcd[bddb]xyyx') == False
        assert is_compatible('aaaa[qwer]tyui') == False
        assert is_compatible('ioxxoj[asdfgh]zxcvbn') == True
        

import unittest

from address import has_reflection, is_compatible, load_addresses


class TestAddress(unittest.TestCase):
    def test_has_reflection(self):
        assert has_reflection(['mnop']) == False
        assert has_reflection(['abba', 'qrst']) == True

    def test_is_compatible(self):
        assert is_compatible('abba[mnop]qrst') == True
        assert is_compatible('abcd[bddb]xyyx') == False
        assert is_compatible('aaaa[qwer]tyui') == False
        assert is_compatible('ioxxoj[asdfgh]zxcvbn') == True
        assert is_compatible('aba[bab]xyz', protocol=2) == True
        assert is_compatible('xyx[xyx]xyx', protocol=2) == False
        assert is_compatible('aaa[kek]eke', protocol=2) == True
        assert is_compatible('zazbz[bzb]cdb', protocol=2) == True
        
    def test_load_addresses(self):
        assert len(load_addresses())

Add tests for second protocol.
Add tests for second protocol.

  import unittest
  
- from address import has_reflection, is_compatible
+ from address import has_reflection, is_compatible, load_addresses
  
  
  class TestAddress(unittest.TestCase):
      def test_has_reflection(self):
          assert has_reflection(['mnop']) == False
          assert has_reflection(['abba', 'qrst']) == True
  
      def test_is_compatible(self):
          assert is_compatible('abba[mnop]qrst') == True
          assert is_compatible('abcd[bddb]xyyx') == False
          assert is_compatible('aaaa[qwer]tyui') == False
          assert is_compatible('ioxxoj[asdfgh]zxcvbn') == True
+         assert is_compatible('aba[bab]xyz', protocol=2) == True
+         assert is_compatible('xyx[xyx]xyx', protocol=2) == False
+         assert is_compatible('aaa[kek]eke', protocol=2) == True
+         assert is_compatible('zazbz[bzb]cdb', protocol=2) == True
          
+     def test_load_addresses(self):
+         assert len(load_addresses())
  
Add tests for second protocol.
## Code Before:
import unittest

from address import has_reflection, is_compatible


class TestAddress(unittest.TestCase):
    def test_has_reflection(self):
        assert has_reflection(['mnop']) == False
        assert has_reflection(['abba', 'qrst']) == True

    def test_is_compatible(self):
        assert is_compatible('abba[mnop]qrst') == True
        assert is_compatible('abcd[bddb]xyyx') == False
        assert is_compatible('aaaa[qwer]tyui') == False
        assert is_compatible('ioxxoj[asdfgh]zxcvbn') == True
        

## Instruction:
Add tests for second protocol.
## Code After:
import unittest

from address import has_reflection, is_compatible, load_addresses


class TestAddress(unittest.TestCase):
    def test_has_reflection(self):
        assert has_reflection(['mnop']) == False
        assert has_reflection(['abba', 'qrst']) == True

    def test_is_compatible(self):
        assert is_compatible('abba[mnop]qrst') == True
        assert is_compatible('abcd[bddb]xyyx') == False
        assert is_compatible('aaaa[qwer]tyui') == False
        assert is_compatible('ioxxoj[asdfgh]zxcvbn') == True
        assert is_compatible('aba[bab]xyz', protocol=2) == True
        assert is_compatible('xyx[xyx]xyx', protocol=2) == False
        assert is_compatible('aaa[kek]eke', protocol=2) == True
        assert is_compatible('zazbz[bzb]cdb', protocol=2) == True
        
    def test_load_addresses(self):
        assert len(load_addresses())
