alex/esp-idf
alex/esp-idf
1
0
Fork 0
mirror of https://github.com/espressif/esp-idf.git synced 2024-10-05 20:47:46 -04:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

tiny-test-fw: fix dut for python2 and python3

Browse Source
This commit is contained in:
houchenyao 2018-01-31 16:57:40 +08:00 committed by bot
parent 475ffe78e8
commit 1e4508bc8d
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tools/tiny-test-fw/DUT.py
View File

@ -316,7 +316,7 @@ class BaseDUT(object):
if flush:
self.data_cache.flush()
# do write if cache
if data:
if data is not None:
self._port_write(data + eol if eol else data)
@_expect_lock
@ -557,9 +557,9 @@ class SerialDUT(BaseDUT):
:return: formatted data (str)
"""
timestamp = time.time()
timestamp = "{}:{}".format(time.strftime("%m-%d %H:%M:%S", time.localtime(timestamp)),
str(timestamp % 1)[2:5])
formatted_data = "[{}]:\r\n{}\r\n".format(timestamp, _decode_data(data))
timestamp = "[{}:{}]".format(time.strftime("%m-%d %H:%M:%S", time.localtime(timestamp)),
str(timestamp % 1)[2:5])
formatted_data = timestamp.encode() + b"\r\n" + data + b"\r\n"
return formatted_data
def _port_open(self):
@ -571,11 +571,13 @@ class SerialDUT(BaseDUT):
def _port_read(self, size=1):
data = self.port_inst.read(size)
if data:
with open(self.log_file, "a+") as _log_file:
with open(self.log_file, "ab+") as _log_file:
_log_file.write(self._format_data(data))
return data
def _port_write(self, data):
if isinstance(data, str):
data = data.encode()
self.port_inst.write(data)
@classmethod