def test_to_text(self): sio = io.BytesIO() sio.write(b"TEST") sio.seek(0) Text.to_txt(sio) test_file_name = glob.glob("*.txt")[0] test_file = open(test_file_name, 'r') line = test_file.read() test_file.close() os.remove(test_file_name) self.assertEqual('TEST', line)
def test_to_stdout(self): test_file = open('test.txt', 'w+') sys.stdout = test_file sio = io.BytesIO() sio.write(b"TEST") sio.seek(0) Text.to_stdout(sio) test_file.seek(0) line = test_file.read() test_file.close() os.remove('test.txt') self.assertEqual('TEST', line)