from .. utils import TranspileTestCase, UnaryOperationTestCase, BinaryOperationTestCase, InplaceOperationTestCase import unittest class RangeTests(TranspileTestCase): def test_lengths(self): self.assertCodeExecution(""" print("len(range(0, 10)) =", len(range(0, 10))) print("len(range(0, 10, 2)) =", len(range(0, 10, 2))) print("len(range(0, 11, 2)) =", len(range(0, 11, 2))) print("len(range(-10, 0)) =", len(range(-10, 0))) print("len(range(10, 0, -1)) =", len(range(10, 0, -1))) print("len(range(10, 0, -2)) =", len(range(10, 0, -2))) print("len(range(11, 0, -2)) =", len(range(11, 0, -2))) print("len(range(0, -10, -1)) =", len(range(0, -10, -1))) print("len(range(0, 5, 2)) =", len(range(0, 5, 2))) print("len(range(7, -6, -3)) =", len(range(7, -6, -3))) print("len(range(1, 10, -1)) =", len(range(1, 10, -1))) print("len(range(10, 0, 1)) =", len(range(10, 0, 1))) """) def test_creation(self): self.assertCodeExecution(""" x = range(0, 5) print("x[0] =", x[0]) print("x[1] =", x[1]) print("x[3] =", x[3]) print("x[-1] =", x[-1]) print("x[5] =", x[5]) """) def test_start_not_zero(self): self.assertCodeExecution(""" x = range(5, 10) print("x[0] =", x[0]) print("x[1] =", x[1]) print("x[3] =", x[3]) print("x[-1] =", x[-1]) print("x[5] =", x[5]) """) def test_step(self): self.assertCodeExecution(""" x = range(0, 5, 2) print("x[0] =", x[0]) print("x[1] =", x[1]) print("x[3] =", x[3]) print("x[-1] =", x[-1]) print("x[5] =", x[5]) """) def test_step_negative(self): self.assertCodeExecution(""" x = range(7, -5, -2) print("x[0] =", x[0]) print("x[1] =", x[1]) print("x[3] =", x[3]) print("x[-1] =", x[-1]) print("x[5] =", x[5]) """) def test_start_negative(self): self.assertCodeExecution(""" x = range(-10, 0) print("x[0] =", x[0]) print("x[1] =", x[1]) print("x[3] =", x[3]) print("x[-1] =", x[-1]) print("x[5] =", x[5]) """) def test_slice_simple(self): self.assertCodeExecution(""" x = range(-10, 10) print("x[0:5] =", x[0:5]) print("x[2:5] =", x[2:5]) print("x[5:10] =", x[5:10]) print("x[12:12] =", x[12:12]) """) def test_slice_step(self): self.assertCodeExecution(""" x = range(-10, 10) print("x[0:5:2] =", x[0:5:2]) print("x[2:5:2] =", x[2:5:2]) print("x[5:10:2] =", x[5:10:2]) print("x[0:5:-2] =", x[0:5:-2]) print("x[2:5:-2] =", x[2:5:-2]) print("x[25:21:-2] =", x[25:21:-2]) """) def test_slice_incomplete(self): self.assertCodeExecution(""" x = range(-10, 10) print("x[:5:2] =", x[:5:2]) print("x[2::2] =", x[2::2]) print("x[::2] =", x[::2]) print("x[:5:-2] =", x[:5:-2]) print("x[2::-2] =", x[2::-2]) print("x[::-2] =", x[::-2]) """) def test_slice_edge_cases_negative(self): self.assertCodeExecution(""" print("range(2, 7, 3)[8:8:-2] =", range(2, 7, 3)[8:8:-2]) print("range(2, 7, 3)[-8:-8:-2] =", range(2, 7, 3)[-8:-8:-2]) print("range(7, 2, 1)[8:8:-2] =", range(7, 2, 1)[8:8:-2]) print("range(7, 2, 1)[-8:-8:-2] =", range(7, 2, 1)[-8:-8:-2]) print("range(7, 2, -1)[8:8:-2] =", range(7, 2, -1)[8:8:-2]) print("range(7, 2, -1)[-8:-8:-2] =", range(7, 2, -1)[-8:-8:-2]) print("range(2, 7, -1)[8:8:-2] =", range(2, 7, -1)[8:8:-2]) print("range(2, 7, -1)[-8:-8:-2] =", range(2, 7, -1)[-8:-8:-2]) """) def test_slice_edge_cases_positive(self): self.assertCodeExecution(""" print("range(2, 7, 2)[8:8] =", range(2, 7, 2)[8:8]) print("range(2, 7, 2)[-8:-8] =", range(2, 7, 2)[-8:-8]) print("range(7, 2, 1)[8:8] =", range(7, 2, 1)[8:8]) print("range(7, 2, 1)[-8:-8] =", range(7, 2, 1)[-8:-8]) print("range(7, 2, -1)[8:8] =", range(7, 2, -1)[8:8]) print("range(7, 2, -1)[-8:-8] =", range(7, 2, -1)[-8:-8]) print("range(2, 7, -1)[8:8] =", range(2, 7, -1)[8:8]) print("range(2, 7, -1)[-8:-8] =", range(2, 7, -1)[-8:-8]) """) def test_slice_edge_cases_incomplete(self): self.assertCodeExecution(""" print("range(2, 7, 2)[::] =", range(2, 7, 2)[::]) print("range(7, 2, 1)[::] =", range(7, 2, 1)[::]) print("range(7, 2, -1)[::] =", range(7, 2, -1)[::]) print("range(2, 7, -1)[::] =", range(2, 7, -1)[::]) print("range(2, 7, 2)[::-1] =", range(2, 7, 2)[::-1]) print("range(7, 2, 1)[::-1] =", range(7, 2, 1)[::-1]) print("range(7, 2, -1)[::-1] =", range(7, 2, -1)[::-1]) print("range(2, 7, -1)[::-1] =", range(2, 7, -1)[::-1]) """) class UnaryRangeOperationTests(UnaryOperationTestCase, TranspileTestCase): data_type = 'range' not_implemented = [ 'test_unary_invert', 'test_unary_negative', 'test_unary_not', 'test_unary_positive', ] class BinaryRangeOperationTests(BinaryOperationTestCase, TranspileTestCase): data_type = 'range' not_implemented = [ 'test_add_bool', 'test_add_bytearray', 'test_add_bytes', 'test_add_class', 'test_add_complex', 'test_add_dict', 'test_add_float', 'test_add_frozenset', 'test_add_int', 'test_add_list', 'test_add_None', 'test_add_NotImplemented', 'test_add_range', 'test_add_set', 'test_add_slice', 'test_add_str', 'test_add_tuple', 'test_and_bool', 'test_and_bytearray', 'test_and_bytes', 'test_and_class', 'test_and_complex', 'test_and_dict', 'test_and_float', 'test_and_frozenset', 'test_and_int', 'test_and_list', 'test_and_None', 'test_and_NotImplemented', 'test_and_range', 'test_and_set', 'test_and_slice', 'test_and_str', 'test_and_tuple', 'test_eq_range', 'test_floor_divide_bool', 'test_floor_divide_bytearray', 'test_floor_divide_bytes', 'test_floor_divide_class', 'test_floor_divide_complex', 'test_floor_divide_dict', 'test_floor_divide_float', 'test_floor_divide_frozenset', 'test_floor_divide_int', 'test_floor_divide_list', 'test_floor_divide_None', 'test_floor_divide_NotImplemented', 'test_floor_divide_range', 'test_floor_divide_set', 'test_floor_divide_slice', 'test_floor_divide_str', 'test_floor_divide_tuple', 'test_ge_bool', 'test_ge_bytearray', 'test_ge_bytes', 'test_ge_class', 'test_ge_complex', 'test_ge_dict', 'test_ge_float', 'test_ge_frozenset', 'test_ge_int', 'test_ge_list', 'test_ge_None', 'test_ge_NotImplemented', 'test_ge_range', 'test_ge_set', 'test_ge_slice', 'test_ge_str', 'test_ge_tuple', 'test_gt_bool', 'test_gt_bytearray', 'test_gt_bytes', 'test_gt_class', 'test_gt_complex', 'test_gt_dict', 'test_gt_float', 'test_gt_frozenset', 'test_gt_int', 'test_gt_list', 'test_gt_None', 'test_gt_NotImplemented', 'test_gt_range', 'test_gt_set', 'test_gt_slice', 'test_gt_str', 'test_gt_tuple', 'test_le_bool', 'test_le_bytearray', 'test_le_bytes', 'test_le_class', 'test_le_complex', 'test_le_dict', 'test_le_float', 'test_le_frozenset', 'test_le_int', 'test_le_list', 'test_le_None', 'test_le_NotImplemented', 'test_le_range', 'test_le_set', 'test_le_slice', 'test_le_str', 'test_le_tuple', 'test_lshift_bool', 'test_lshift_bytearray', 'test_lshift_bytes', 'test_lshift_class', 'test_lshift_complex', 'test_lshift_dict', 'test_lshift_float', 'test_lshift_frozenset', 'test_lshift_int', 'test_lshift_list', 'test_lshift_None', 'test_lshift_NotImplemented', 'test_lshift_range', 'test_lshift_set', 'test_lshift_slice', 'test_lshift_str', 'test_lshift_tuple', 'test_lt_bool', 'test_lt_bytearray', 'test_lt_bytes', 'test_lt_class', 'test_lt_complex', 'test_lt_dict', 'test_lt_float', 'test_lt_frozenset', 'test_lt_int', 'test_lt_list', 'test_lt_None', 'test_lt_NotImplemented', 'test_lt_range', 'test_lt_set', 'test_lt_slice', 'test_lt_str', 'test_lt_tuple', 'test_modulo_bool', 'test_modulo_bytearray', 'test_modulo_bytes', 'test_modulo_class', 'test_modulo_complex', 'test_modulo_dict', 'test_modulo_float', 'test_modulo_frozenset', 'test_modulo_int', 'test_modulo_list', 'test_modulo_None', 'test_modulo_NotImplemented', 'test_modulo_range', 'test_modulo_set', 'test_modulo_slice', 'test_modulo_str', 'test_modulo_tuple', 'test_multiply_bool', 'test_multiply_bytearray', 'test_multiply_bytes', 'test_multiply_class', 'test_multiply_complex', 'test_multiply_dict', 'test_multiply_float', 'test_multiply_frozenset', 'test_multiply_int', 'test_multiply_list', 'test_multiply_None', 'test_multiply_NotImplemented', 'test_multiply_range', 'test_multiply_set', 'test_multiply_slice', 'test_multiply_str', 'test_multiply_tuple', 'test_ne_range', 'test_or_bool', 'test_or_bytearray', 'test_or_bytes', 'test_or_class', 'test_or_complex', 'test_or_dict', 'test_or_float', 'test_or_frozenset', 'test_or_int', 'test_or_list', 'test_or_None', 'test_or_NotImplemented', 'test_or_range', 'test_or_set', 'test_or_slice', 'test_or_str', 'test_or_tuple', 'test_power_bool', 'test_power_bytearray', 'test_power_bytes', 'test_power_class', 'test_power_complex', 'test_power_dict', 'test_power_float', 'test_power_frozenset', 'test_power_int', 'test_power_list', 'test_power_None', 'test_power_NotImplemented', 'test_power_range', 'test_power_set', 'test_power_slice', 'test_power_str', 'test_power_tuple', 'test_rshift_bool', 'test_rshift_bytearray', 'test_rshift_bytes', 'test_rshift_class', 'test_rshift_complex', 'test_rshift_dict', 'test_rshift_float', 'test_rshift_frozenset', 'test_rshift_int', 'test_rshift_list', 'test_rshift_None', 'test_rshift_NotImplemented', 'test_rshift_range', 'test_rshift_set', 'test_rshift_slice', 'test_rshift_str', 'test_rshift_tuple', 'test_subtract_bool', 'test_subtract_bytearray', 'test_subtract_bytes', 'test_subtract_class', 'test_subtract_complex', 'test_subtract_dict', 'test_subtract_float', 'test_subtract_frozenset', 'test_subtract_int', 'test_subtract_list', 'test_subtract_None', 'test_subtract_NotImplemented', 'test_subtract_range', 'test_subtract_set', 'test_subtract_slice', 'test_subtract_str', 'test_subtract_tuple', 'test_true_divide_bool', 'test_true_divide_bytearray', 'test_true_divide_bytes', 'test_true_divide_class', 'test_true_divide_complex', 'test_true_divide_dict', 'test_true_divide_float', 'test_true_divide_frozenset', 'test_true_divide_int', 'test_true_divide_list', 'test_true_divide_None', 'test_true_divide_NotImplemented', 'test_true_divide_range', 'test_true_divide_set', 'test_true_divide_slice', 'test_true_divide_str', 'test_true_divide_tuple', 'test_xor_bool', 'test_xor_bytearray', 'test_xor_bytes', 'test_xor_class', 'test_xor_complex', 'test_xor_dict', 'test_xor_float', 'test_xor_frozenset', 'test_xor_int', 'test_xor_list', 'test_xor_None', 'test_xor_NotImplemented', 'test_xor_range', 'test_xor_set', 'test_xor_slice', 'test_xor_str', 'test_xor_tuple', ] class InplaceRangeOperationTests(InplaceOperationTestCase, TranspileTestCase): data_type = 'range' not_implemented = [ 'test_add_bool', 'test_add_bytearray', 'test_add_bytes', 'test_add_class', 'test_add_complex', 'test_add_dict', 'test_add_float', 'test_add_frozenset', 'test_add_int', 'test_add_list', 'test_add_None', 'test_add_NotImplemented', 'test_add_range', 'test_add_set', 'test_add_slice', 'test_add_str', 'test_add_tuple', 'test_and_bool', 'test_and_bytearray', 'test_and_bytes', 'test_and_class', 'test_and_complex', 'test_and_dict', 'test_and_float', 'test_and_frozenset', 'test_and_int', 'test_and_list', 'test_and_None', 'test_and_NotImplemented', 'test_and_range', 'test_and_set', 'test_and_slice', 'test_and_str', 'test_and_tuple', 'test_floor_divide_bool', 'test_floor_divide_bytearray', 'test_floor_divide_bytes', 'test_floor_divide_class', 'test_floor_divide_complex', 'test_floor_divide_dict', 'test_floor_divide_float', 'test_floor_divide_frozenset', 'test_floor_divide_int', 'test_floor_divide_list', 'test_floor_divide_None', 'test_floor_divide_NotImplemented', 'test_floor_divide_range', 'test_floor_divide_set', 'test_floor_divide_slice', 'test_floor_divide_str', 'test_floor_divide_tuple', 'test_lshift_bool', 'test_lshift_bytearray', 'test_lshift_bytes', 'test_lshift_class', 'test_lshift_complex', 'test_lshift_dict', 'test_lshift_float', 'test_lshift_frozenset', 'test_lshift_int', 'test_lshift_list', 'test_lshift_None', 'test_lshift_NotImplemented', 'test_lshift_range', 'test_lshift_set', 'test_lshift_slice', 'test_lshift_str', 'test_lshift_tuple', 'test_modulo_bool', 'test_modulo_bytearray', 'test_modulo_bytes', 'test_modulo_class', 'test_modulo_complex', 'test_modulo_dict', 'test_modulo_float', 'test_modulo_frozenset', 'test_modulo_int', 'test_modulo_list', 'test_modulo_None', 'test_modulo_NotImplemented', 'test_modulo_range', 'test_modulo_set', 'test_modulo_slice', 'test_modulo_str', 'test_modulo_tuple', 'test_multiply_bool', 'test_multiply_bytearray', 'test_multiply_bytes', 'test_multiply_class', 'test_multiply_complex', 'test_multiply_dict', 'test_multiply_float', 'test_multiply_frozenset', 'test_multiply_int', 'test_multiply_list', 'test_multiply_None', 'test_multiply_NotImplemented', 'test_multiply_range', 'test_multiply_set', 'test_multiply_slice', 'test_multiply_str', 'test_multiply_tuple', 'test_or_bool', 'test_or_bytearray', 'test_or_bytes', 'test_or_class', 'test_or_complex', 'test_or_dict', 'test_or_float', 'test_or_frozenset', 'test_or_int', 'test_or_list', 'test_or_None', 'test_or_NotImplemented', 'test_or_range', 'test_or_set', 'test_or_slice', 'test_or_str', 'test_or_tuple', 'test_power_bool', 'test_power_bytearray', 'test_power_bytes', 'test_power_class', 'test_power_complex', 'test_power_dict', 'test_power_float', 'test_power_frozenset', 'test_power_int', 'test_power_list', 'test_power_None', 'test_power_NotImplemented', 'test_power_range', 'test_power_set', 'test_power_slice', 'test_power_str', 'test_power_tuple', 'test_rshift_bool', 'test_rshift_bytearray', 'test_rshift_bytes', 'test_rshift_class', 'test_rshift_complex', 'test_rshift_dict', 'test_rshift_float', 'test_rshift_frozenset', 'test_rshift_int', 'test_rshift_list', 'test_rshift_None', 'test_rshift_NotImplemented', 'test_rshift_range', 'test_rshift_set', 'test_rshift_slice', 'test_rshift_str', 'test_rshift_tuple', 'test_subtract_bool', 'test_subtract_bytearray', 'test_subtract_bytes', 'test_subtract_class', 'test_subtract_complex', 'test_subtract_dict', 'test_subtract_float', 'test_subtract_frozenset', 'test_subtract_int', 'test_subtract_list', 'test_subtract_None', 'test_subtract_NotImplemented', 'test_subtract_range', 'test_subtract_set', 'test_subtract_slice', 'test_subtract_str', 'test_subtract_tuple', 'test_true_divide_bool', 'test_true_divide_bytearray', 'test_true_divide_bytes', 'test_true_divide_class', 'test_true_divide_complex', 'test_true_divide_dict', 'test_true_divide_float', 'test_true_divide_frozenset', 'test_true_divide_int', 'test_true_divide_list', 'test_true_divide_None', 'test_true_divide_NotImplemented', 'test_true_divide_range', 'test_true_divide_set', 'test_true_divide_slice', 'test_true_divide_str', 'test_true_divide_tuple', 'test_xor_bool', 'test_xor_bytearray', 'test_xor_bytes', 'test_xor_class', 'test_xor_complex', 'test_xor_dict', 'test_xor_float', 'test_xor_frozenset', 'test_xor_int', 'test_xor_list', 'test_xor_None', 'test_xor_NotImplemented', 'test_xor_range', 'test_xor_set', 'test_xor_slice', 'test_xor_str', 'test_xor_tuple', ]