From 9ed4e9c1cdb5963f1bf6d9e4eef3ce0ca54b6e4d Mon Sep 17 00:00:00 2001 From: whitequark Date: Fri, 9 Jun 2017 07:04:14 +0000 Subject: [PATCH] compiler: add support for printing of bytearray values (#714). --- artiq/compiler/transforms/artiq_ir_generator.py | 5 ++++- artiq/test/lit/integration/print.py | 3 +++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/artiq/compiler/transforms/artiq_ir_generator.py b/artiq/compiler/transforms/artiq_ir_generator.py index 1d2f6cbba..9a580f6fb 100644 --- a/artiq/compiler/transforms/artiq_ir_generator.py +++ b/artiq/compiler/transforms/artiq_ir_generator.py @@ -1989,6 +1989,8 @@ class ARTIQIRGenerator(algorithm.Visitor): format_string += "["; flush() elif builtins.is_bytes(value.type): format_string += "bytes(["; flush() + elif builtins.is_bytearray(value.type): + format_string += "bytearray(["; flush() elif builtins.is_array(value.type): format_string += "array(["; flush() else: @@ -2017,7 +2019,8 @@ class ARTIQIRGenerator(algorithm.Visitor): if builtins.is_list(value.type): format_string += "]" - elif builtins.is_array(value.type) or builtins.is_bytes(value.type): + elif (builtins.is_bytes(value.type) or builtins.is_bytearray(value.type) or + builtins.is_array(value.type)): format_string += "])" elif builtins.is_range(value.type): format_string += "range("; flush() diff --git a/artiq/test/lit/integration/print.py b/artiq/test/lit/integration/print.py index bd5bf555f..11887bf77 100644 --- a/artiq/test/lit/integration/print.py +++ b/artiq/test/lit/integration/print.py @@ -36,3 +36,6 @@ print(array([1, 2])) # CHECK-L: bytes([97, 98]) print(b"ab") + +# CHECK-L: bytearray([97, 98]) +print(bytearray(b"ab"))