from .. utils import TranspileTestCase, BuiltinTwoargFunctionTestCase

from unittest import expectedFailure


class MapTests(TranspileTestCase):
    base_code = """
            x = %s
            def testish(x):
                return %s

            print(map(testish, x))
            mylist = iter(x)
            print(map(testish, mylist).__next__())
            print(map(testish, mylist).__next__())
            print(map(testish, mylist).__next__())
            try:
                print(map(testish, mylist).__next__())
            except StopIteration:
                pass
    """

    def test_bool(self):
        self.assertCodeExecution(self.base_code % ("[True, False, True]", "bool(x)"))

    def test_bytearray(self):
        self.assertCodeExecution(self.base_code % ("b'123'", "x"))

    def test_float(self):
        self.assertCodeExecution(self.base_code % ("[3.14, 2.17, 1.0]", "x > 1"))

    def test_int(self):
        self.assertCodeExecution(self.base_code % ("[1, 2, 3]", "x * 2"))


class BuiltinMapFunctionTests(BuiltinTwoargFunctionTestCase, TranspileTestCase):
    functions = ["map"]

    not_implemented = [
        'test_bool_bool',
        'test_bool_class',
        'test_bool_complex',
        'test_bool_float',
        'test_bool_int',
        'test_bool_None',
        'test_bool_NotImplemented',
        'test_bool_slice',

        'test_bytearray_bool',
        'test_bytearray_class',
        'test_bytearray_complex',
        'test_bytearray_float',
        'test_bytearray_int',
        'test_bytearray_None',
        'test_bytearray_NotImplemented',
        'test_bytearray_slice',

        'test_bytes_bool',
        'test_bytes_class',
        'test_bytes_complex',
        'test_bytes_float',
        'test_bytes_int',
        'test_bytes_None',
        'test_bytes_NotImplemented',
        'test_bytes_slice',

        'test_class_bool',
        'test_class_class',
        'test_class_complex',
        'test_class_float',
        'test_class_int',
        'test_class_None',
        'test_class_NotImplemented',
        'test_class_slice',

        'test_complex_bool',
        'test_complex_class',
        'test_complex_complex',
        'test_complex_float',
        'test_complex_int',
        'test_complex_None',
        'test_complex_NotImplemented',
        'test_complex_slice',

        'test_dict_bool',
        'test_dict_class',
        'test_dict_complex',
        'test_dict_float',
        'test_dict_int',
        'test_dict_None',
        'test_dict_NotImplemented',
        'test_dict_slice',

        'test_float_bool',
        'test_float_class',
        'test_float_complex',
        'test_float_float',
        'test_float_int',
        'test_float_None',
        'test_float_NotImplemented',
        'test_float_slice',

        'test_frozenset_bool',
        'test_frozenset_class',
        'test_frozenset_complex',
        'test_frozenset_float',
        'test_frozenset_int',
        'test_frozenset_None',
        'test_frozenset_NotImplemented',
        'test_frozenset_slice',

        'test_int_bool',
        'test_int_class',
        'test_int_complex',
        'test_int_float',
        'test_int_int',
        'test_int_None',
        'test_int_NotImplemented',
        'test_int_slice',

        'test_list_bool',
        'test_list_class',
        'test_list_complex',
        'test_list_float',
        'test_list_int',
        'test_list_None',
        'test_list_NotImplemented',
        'test_list_slice',

        'test_None_bool',
        'test_None_class',
        'test_None_complex',
        'test_None_float',
        'test_None_int',
        'test_None_None',
        'test_None_NotImplemented',
        'test_None_slice',

        'test_NotImplemented_bool',
        'test_NotImplemented_class',
        'test_NotImplemented_complex',
        'test_NotImplemented_float',
        'test_NotImplemented_int',
        'test_NotImplemented_None',
        'test_NotImplemented_NotImplemented',
        'test_NotImplemented_slice',

        'test_range_bool',
        'test_range_class',
        'test_range_complex',
        'test_range_float',
        'test_range_int',
        'test_range_None',
        'test_range_NotImplemented',
        'test_range_slice',

        'test_set_bool',
        'test_set_class',
        'test_set_complex',
        'test_set_float',
        'test_set_int',
        'test_set_None',
        'test_set_NotImplemented',
        'test_set_slice',

        'test_slice_bool',
        'test_slice_class',
        'test_slice_complex',
        'test_slice_float',
        'test_slice_int',
        'test_slice_None',
        'test_slice_NotImplemented',
        'test_slice_slice',

        'test_str_bool',
        'test_str_class',
        'test_str_complex',
        'test_str_float',
        'test_str_int',
        'test_str_None',
        'test_str_NotImplemented',
        'test_str_slice',

        'test_tuple_bool',
        'test_tuple_class',
        'test_tuple_complex',
        'test_tuple_float',
        'test_tuple_int',
        'test_tuple_None',
        'test_tuple_NotImplemented',
        'test_tuple_slice',
    ]