Beispiel #1
0
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
Beispiel #2
0
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