from unittest import TestCase from stackify.compat import b from stackify.compat import iterkeys from stackify.compat import iteritems class CompatTest(TestCase): def setUp(self): self.dict_data = { "key1": "value1", "key2": "value2" } self.list_data = ["foo", "bar"] def test_convert_string_to_byte(self): byte = '1' value = b(byte) assert isinstance(value, bytes) def test_iterkeys_should_return_iterator(self): iter_keys = iterkeys(self.dict_data) self.assert_instance_is_an_iterator(iter_keys) def test_iteritems_should_return_iterator(self): iter_items = iteritems(self.dict_data) self.assert_instance_is_an_iterator(iter_items) def assert_instance_is_an_iterator(self, item): try: iter(item) assert True except Exception: assert False