def test(): """Ensure that calling the buffer is not "flushed" between put calls""" n = Writer(filepath) byte_count = 0 n.put_data([10, 10], item_size=5) n.put_data([10, 10], item_size=5) n._flush_buffer() n.close() fh = open(filepath, "rb") for byte in fh.read(): byte_count += 1 fh.close() os.remove(filepath) return byte_count == 3
def test(): """Basic test to ensure that writing 5 bits at a time works as expected""" n = Writer(filepath) data_good = True test_bytes = [74, 82, 144] byte_index = 0 n.put_data([9, 9, 9, 9], item_size=5) n._flush_buffer() n.close() fh = open(filepath, "rb") for byte in fh.read(): data_good &= ord(byte) == test_bytes[byte_index] if not data_good: break byte_index += 1 fh.close() os.remove(filepath) return data_good