Add tests based on TestAllTypes from the C++ test.capnp. Fix problems uncovered in capnp.pyx. Requires capnproto-c++-0.3.0-rc5, as I found a bug in the base library.

2013-08-31 16:44:54 -07:00
parent 10c6f40c2e
commit 0478ee3081
7 changed files with 420 additions and 16 deletions
--- a/test/test_regression.py
+++ b/test/test_regression.py
@@ -1,6 +1,7 @@
 import pytest
 import capnp
 import os
+import math

 this_dir = os.path.dirname(__file__)

@@ -65,3 +66,226 @@ def test_addressbook(addressbook):

    f = open('example', 'r')
    printAddressBook(f.fileno())
+
+@pytest.fixture
+def all_types():
+    return capnp.load(os.path.join(this_dir, 'all-types.capnp'))
+
+# TODO:  These tests should be extended to:
+# - Read each field in Python and assert that it is equal to the expected value.
+# - Build an identical message using Python code and compare it to the golden.
+#
+
+def init_all_types(builder):
+    builder.voidField = None
+    builder.boolField = True
+    builder.int8Field = -123
+    builder.int16Field = -12345
+    builder.int32Field = -12345678
+    builder.int64Field = -123456789012345
+    builder.uInt8Field = 234
+    builder.uInt16Field = 45678
+    builder.uInt32Field = 3456789012
+    builder.uInt64Field = 12345678901234567890
+    builder.float32Field = 1234.5
+    builder.float64Field = -123e45
+    builder.textField = "foo"
+    builder.dataField = "bar"
+
+    subBuilder = builder.structField
+    subBuilder.voidField = None
+    subBuilder.boolField = True
+    subBuilder.int8Field = -12
+    subBuilder.int16Field = 3456
+    subBuilder.int32Field = -78901234
+    subBuilder.int64Field = 56789012345678
+    subBuilder.uInt8Field = 90
+    subBuilder.uInt16Field = 1234
+    subBuilder.uInt32Field = 56789012
+    subBuilder.uInt64Field = 345678901234567890
+    subBuilder.float32Field = -1.25e-10
+    subBuilder.float64Field = 345
+    subBuilder.textField = "baz"
+    subBuilder.dataField = "qux"
+    subSubBuilder = subBuilder.structField
+    subSubBuilder.textField = "nested"
+    subSubBuilder.structField.textField = "really nested"
+    subBuilder.enumField = "baz"
+
+    subBuilder.voidList = [None, None, None]
+    subBuilder.boolList = [False, True, False, True, True]
+    subBuilder.int8List = [12, -34, -0x80, 0x7f]
+    subBuilder.int16List = [1234, -5678, -0x8000, 0x7fff]
+    subBuilder.int32List = [12345678, -90123456, -0x80000000, 0x7fffffff]
+    subBuilder.int64List = [123456789012345, -678901234567890, -0x8000000000000000, 0x7fffffffffffffff]
+    subBuilder.uInt8List = [12, 34, 0, 0xff]
+    subBuilder.uInt16List = [1234, 5678, 0, 0xffff]
+    subBuilder.uInt32List = [12345678, 90123456, 0, 0xffffffff]
+    subBuilder.uInt64List = [123456789012345, 678901234567890, 0, 0xffffffffffffffff]
+    subBuilder.float32List = [0, 1234567, 1e37, -1e37, 1e-37, -1e-37]
+    subBuilder.float64List = [0, 123456789012345, 1e306, -1e306, 1e-306, -1e-306]
+    subBuilder.textList = ["quux", "corge", "grault"]
+    subBuilder.dataList = ["garply", "waldo", "fred"]
+    listBuilder = subBuilder.init('structList', 3)
+    listBuilder[0].textField = "x structlist 1"
+    listBuilder[1].textField = "x structlist 2"
+    listBuilder[2].textField = "x structlist 3"
+    subBuilder.enumList = ["qux", "bar", "grault"]
+
+    builder.enumField = "corge"
+
+    builder.init("voidList", 6)
+    builder.boolList = [True, False, False, True]
+    builder.int8List = [111, -111]
+    builder.int16List = [11111, -11111]
+    builder.int32List = [111111111, -111111111]
+    builder.int64List = [1111111111111111111, -1111111111111111111]
+    builder.uInt8List = [111, 222]
+    builder.uInt16List = [33333, 44444]
+    builder.uInt32List = [3333333333]
+    builder.uInt64List = [11111111111111111111]
+    builder.float32List = [5555.5, float("inf"), float("-inf"), float("nan")]
+    builder.float64List = [7777.75, float("inf"), float("-inf"), float("nan")]
+    builder.textList = ["plugh", "xyzzy", "thud"]
+    builder.dataList = ["oops", "exhausted", "rfc3092"]
+    listBuilder = builder.init('structList', 3)
+    listBuilder[0].textField = "structlist 1"
+    listBuilder[1].textField = "structlist 2"
+    listBuilder[2].textField = "structlist 3"
+    builder.enumList = ["foo", "garply"]
+
+def assert_almost(float1, float2):
+    if float1 != float2:
+        assert abs((float1 - float2) / float1) < 0.00001
+
+def check_list(reader, expected):
+    assert len(reader) == len(expected)
+    for (i, v) in enumerate(expected):
+        if type(v) is float:
+            assert_almost(reader[i], v)
+        else:
+            assert reader[i] == v
+
+def check_all_types(reader):
+    assert reader.voidField == None
+    assert reader.boolField == True
+    assert reader.int8Field == -123
+    assert reader.int16Field == -12345
+    assert reader.int32Field == -12345678
+    assert reader.int64Field == -123456789012345
+    assert reader.uInt8Field == 234
+    assert reader.uInt16Field == 45678
+    assert reader.uInt32Field == 3456789012
+    assert reader.uInt64Field == 12345678901234567890
+    assert reader.float32Field == 1234.5
+    assert_almost(reader.float64Field, -123e45)
+    assert reader.textField == "foo"
+    assert reader.dataField == "bar"
+
+    subReader = reader.structField
+    assert subReader.voidField == None
+    assert subReader.boolField == True
+    assert subReader.int8Field == -12
+    assert subReader.int16Field == 3456
+    assert subReader.int32Field == -78901234
+    assert subReader.int64Field == 56789012345678
+    assert subReader.uInt8Field == 90
+    assert subReader.uInt16Field == 1234
+    assert subReader.uInt32Field == 56789012
+    assert subReader.uInt64Field == 345678901234567890
+    assert_almost(subReader.float32Field, -1.25e-10)
+    assert subReader.float64Field == 345
+    assert subReader.textField == "baz"
+    assert subReader.dataField == "qux"
+
+    subSubReader = subReader.structField
+    assert subSubReader.textField == "nested"
+    assert subSubReader.structField.textField == "really nested"
+
+    assert subReader.enumField == "baz"
+
+    check_list(subReader.voidList, [None, None, None])
+    check_list(subReader.boolList, [False, True, False, True, True])
+    check_list(subReader.int8List, [12, -34, -0x80, 0x7f])
+    check_list(subReader.int16List, [1234, -5678, -0x8000, 0x7fff])
+    check_list(subReader.int32List, [12345678, -90123456, -0x80000000, 0x7fffffff])
+    check_list(subReader.int64List, [123456789012345, -678901234567890, -0x8000000000000000, 0x7fffffffffffffff])
+    check_list(subReader.uInt8List, [12, 34, 0, 0xff])
+    check_list(subReader.uInt16List, [1234, 5678, 0, 0xffff])
+    check_list(subReader.uInt32List, [12345678, 90123456, 0, 0xffffffff])
+    check_list(subReader.uInt64List, [123456789012345, 678901234567890, 0, 0xffffffffffffffff])
+    check_list(subReader.float32List, [0.0, 1234567.0, 1e37, -1e37, 1e-37, -1e-37])
+    check_list(subReader.float64List, [0.0, 123456789012345.0, 1e306, -1e306, 1e-306, -1e-306])
+    check_list(subReader.textList, ["quux", "corge", "grault"])
+    check_list(subReader.dataList, ["garply", "waldo", "fred"])
+
+    listReader = subReader.structList
+    assert len(listReader) == 3
+    assert listReader[0].textField == "x structlist 1"
+    assert listReader[1].textField == "x structlist 2"
+    assert listReader[2].textField == "x structlist 3"
+
+    check_list(subReader.enumList, ["qux", "bar", "grault"])
+
+    assert reader.enumField == "corge"
+
+    assert len(reader.voidList) == 6
+    check_list(reader.boolList, [True, False, False, True])
+    check_list(reader.int8List, [111, -111])
+    check_list(reader.int16List, [11111, -11111])
+    check_list(reader.int32List, [111111111, -111111111])
+    check_list(reader.int64List, [1111111111111111111, -1111111111111111111])
+    check_list(reader.uInt8List, [111, 222])
+    check_list(reader.uInt16List, [33333, 44444])
+    check_list(reader.uInt32List, [3333333333])
+    check_list(reader.uInt64List, [11111111111111111111])
+
+    listReader = reader.float32List
+    assert len(listReader) == 4
+    assert listReader[0] == 5555.5
+    assert listReader[1] == float("inf")
+    assert listReader[2] == -float("inf")
+    assert math.isnan(listReader[3])
+
+    listReader = reader.float64List
+    len(listReader) == 4
+    assert listReader[0] == 7777.75
+    assert listReader[1] == float("inf")
+    assert listReader[2] == -float("inf")
+    assert math.isnan(listReader[3])
+
+    check_list(reader.textList, ["plugh", "xyzzy", "thud"])
+    check_list(reader.dataList, ["oops", "exhausted", "rfc3092"])
+
+    listReader = reader.structList
+    len(listReader) == 3
+    assert listReader[0].textField == "structlist 1"
+    assert listReader[1].textField == "structlist 2"
+    assert listReader[2].textField == "structlist 3"
+
+    check_list(reader.enumList, ["foo", "garply"])
+
+def test_build(all_types):
+    builder = capnp.MallocMessageBuilder()
+    root = builder.getRoot(all_types.TestAllTypes)
+    init_all_types(root)
+    expectedText = open(os.path.join(this_dir, 'all-types.txt'), 'r').read()
+    assert str(root) + '\n' == expectedText
+
+def test_binary_read(all_types):
+    f = open(os.path.join(this_dir, 'all-types.binary'), 'r')
+    message = capnp.StreamFdMessageReader(f.fileno())
+    root = message.getRoot(all_types.TestAllTypes)
+    check_all_types(root)
+
+    expectedText = open(os.path.join(this_dir, 'all-types.txt'), 'r').read()
+    assert str(root) + '\n' == expectedText
+
+def test_packed_read(all_types):
+    f = open(os.path.join(this_dir, 'all-types.packed'), 'r')
+    message = capnp.PackedFdMessageReader(f.fileno())
+    root = message.getRoot(all_types.TestAllTypes)
+    check_all_types(root)
+
+    expectedText = open(os.path.join(this_dir, 'all-types.txt'), 'r').read()
+    assert str(root) + '\n' == expectedText