diff options
Diffstat (limited to 'tests/luatest.lua')
-rw-r--r-- | tests/luatest.lua | 294 |
1 files changed, 294 insertions, 0 deletions
diff --git a/tests/luatest.lua b/tests/luatest.lua new file mode 100644 index 00000000..23bc0f29 --- /dev/null +++ b/tests/luatest.lua @@ -0,0 +1,294 @@ +package.path = string.format("../lua/?.lua;./?.lua;%s",package.path) + +local function checkReadBuffer(buf, offset, sizePrefix) + offset = offset or 0 + + if type(buf) == "string" then + buf = flatbuffers.binaryArray.New(buf) + end + + if sizePrefix then + local size = flatbuffers.N.Int32:Unpack(buf, offset) + -- no longer matches python tests, but the latest 'monsterdata_test.mon' + -- is 448 bytes, minus 4 to arrive at the 444 + assert(size == 444) + offset = offset + flatbuffers.N.Int32.bytewidth + end + + local mon = monster.GetRootAsMonster(buf, offset) + assert(mon:Hp() == 80, "Monster Hp is not 80") + assert(mon:Mana() == 150, "Monster Mana is not 150") + assert(mon:Name() == "MyMonster", "Monster Name is not MyMonster") + + local vec = assert(mon:Pos(), "Monster Position is nil") + assert(vec:X() == 1.0) + assert(vec:Y() == 2.0) + assert(vec:Z() == 3.0) + assert(vec:Test1() == 3.0) + assert(vec:Test2() == 2) + + local t = require("MyGame.Example.Test").New() + t = assert(vec:Test3(t)) + + assert(t:A() == 5) + assert(t:B() == 6) + + local ut = require("MyGame.Example.Any") + assert(mon:TestType() == ut.Monster) + + local table2 = mon:Test() + assert(getmetatable(table2) == "flatbuffers.view.mt") + + local mon2 = monster.New() + mon2:Init(table2.bytes, table2.pos) + + assert(mon2:Name() == "Fred") + + assert(mon:InventoryLength() == 5) + local invsum = 0 + for i=1,mon:InventoryLength() do + local v = mon:Inventory(i) + invsum = invsum + v + end + assert(invsum == 10) + + for i=1,5 do + assert(mon:VectorOfLongs(i) == 10^((i-1)*2)) + end + + local dbls = { -1.7976931348623157e+308, 0, 1.7976931348623157e+308} + for i=1,mon:VectorOfDoublesLength() do + assert(mon:VectorOfDoubles(i) == dbls[i]) + end + + assert(mon:Test4Length() == 2) + + local test0 = mon:Test4(1) + local test1 = mon:Test4(2) + + local v0 = test0:A() + local v1 = test0:B() + local v2 = test1:A() + local v3 = test1:B() + + local sumtest12 = v0 + v1 + v2 + v3 + assert(sumtest12 == 100) + + assert(mon:TestarrayofstringLength() == 2) + assert(mon:Testarrayofstring(1) == "test1") + assert(mon:Testarrayofstring(2) == "test2") + + assert(mon:TestarrayoftablesLength() == 0) + assert(mon:TestnestedflatbufferLength() == 0) + assert(mon:Testempty() == nil) +end + +local function generateMonster(sizePrefix) + local b = flatbuffers.Builder(0) + local str = b:CreateString("MyMonster") + local test1 = b:CreateString("test1") + local test2 = b:CreateString("test2") + local fred = b:CreateString("Fred") + + monster.StartInventoryVector(b, 5) + b:PrependByte(4) + b:PrependByte(3) + b:PrependByte(2) + b:PrependByte(1) + b:PrependByte(0) + local inv = b:EndVector(5) + + monster.Start(b) + monster.AddName(b, fred) + local mon2 = monster.End(b) + + monster.StartTest4Vector(b, 2) + test.CreateTest(b, 10, 20) + test.CreateTest(b, 30, 40) + local test4 = b:EndVector(2) + + monster.StartTestarrayofstringVector(b, 2) + b:PrependUOffsetTRelative(test2) + b:PrependUOffsetTRelative(test1) + local testArrayOfString = b:EndVector(2) + + monster.StartVectorOfLongsVector(b, 5) + b:PrependInt64(100000000) + b:PrependInt64(1000000) + b:PrependInt64(10000) + b:PrependInt64(100) + b:PrependInt64(1) + local vectorOfLongs = b:EndVector(5) + + monster.StartVectorOfDoublesVector(b, 3) + b:PrependFloat64(1.7976931348623157e+308) + b:PrependFloat64(0) + b:PrependFloat64(-1.7976931348623157e+308) + local vectorOfDoubles = b:EndVector(3) + + monster.Start(b) + local pos = vec3.CreateVec3(b, 1.0, 2.0, 3.0, 3.0, 2, 5, 6) + monster.AddPos(b, pos) + + monster.AddHp(b, 80) + monster.AddName(b, str) + monster.AddInventory(b, inv) + monster.AddTestType(b, 1) + monster.AddTest(b, mon2) + monster.AddTest4(b, test4) + monster.AddTestarrayofstring(b, testArrayOfString) + monster.AddVectorOfLongs(b, vectorOfLongs) + monster.AddVectorOfDoubles(b, vectorOfDoubles) + local mon = monster.End(b) + + if sizePrefix then + b:FinishSizePrefixed(mon) + else + b:Finish(mon) + end + return b:Output(true), b:Head() +end + +local function sizePrefix(sizePrefix) + local buf,offset = generateMonster(sizePrefix) + checkReadBuffer(buf, offset, sizePrefix) +end + +local function testCanonicalData() + local f = assert(io.open('monsterdata_test.mon', 'rb')) + local wireData = f:read("*a") + f:close() + checkReadBuffer(wireData) +end + +local function benchmarkMakeMonster(count) + local length = #(generateMonster()) + + --require("flatbuffers.profiler") + --profiler = newProfiler("call") + --profiler:start() + + local s = os.clock() + for i=1,count do + generateMonster() + end + local e = os.clock() + + --profiler:stop() + + --local outfile = io.open( "profile.txt", "w+" ) + --profiler:report( outfile, true) + --outfile:close() + + + local dur = (e - s) + local rate = count / (dur * 1000) + local data = (length * count) / (1024 * 1024) + local dataRate = data / dur + + print(string.format('built %d %d-byte flatbuffers in %.2fsec: %.2f/msec, %.2fMB/sec', + count, length, dur, rate, dataRate)) +end + +local function benchmarkReadBuffer(count) + local f = assert(io.open('monsterdata_test.mon', 'rb')) + local buf = f:read("*a") + f:close() + + local s = os.clock() + for i=1,count do + checkReadBuffer(buf) + end + local e = os.clock() + + local dur = (e - s) + local rate = count / (dur * 1000) + local data = (#buf * count) / (1024 * 1024) + local dataRate = data / dur + + print(string.format('traversed %d %d-byte flatbuffers in %.2fsec: %.2f/msec, %.2fMB/sec', + count, #buf, dur, rate, dataRate)) +end + +local tests = +{ + { + f = sizePrefix, + d = "Test size prefix", + args = {{true}, {false}} + }, + { + f = testCanonicalData, + d = "Tests Canonical flatbuffer file included in repo" + }, + { + f = benchmarkMakeMonster, + d = "Benchmark making monsters", + args = { + {100}, + {1000}, + {10000}, + } + }, + { + f = benchmarkReadBuffer, + d = "Benchmark reading monsters", + args = { + {100}, + {1000}, + {10000}, + -- uncomment following to run 1 million to compare. + -- Took ~141 seconds on my machine + --{1000000}, + } + }, +} + +local result, err = xpcall(function() + flatbuffers = assert(require("flatbuffers")) + monster = assert(require("MyGame.Example.Monster")) + test = assert(require("MyGame.Example.Test")) + vec3 = assert(require("MyGame.Example.Vec3")) + + local function buildArgList(tbl) + local s = "" + for _,item in ipairs(tbl) do + s = s .. tostring(item) .. "," + end + return s:sub(1,-2) + end + + local testsPassed, testsFailed = 0,0 + for _,test in ipairs(tests) do + local allargs = test.args or {{}} + for _,args in ipairs(allargs) do + local results, err = xpcall(test.f,debug.traceback, table.unpack(args)) + if results then + testsPassed = testsPassed + 1 + else + testsFailed = testsFailed + 1 + print(string.format(" Test [%s](%s) failed: \n\t%s", + test.d or "", + buildArgList(args), + err)) + end + end + end + + local totalTests = testsPassed + testsFailed + print(string.format("# of test passed: %d / %d (%.2f%%)", + testsPassed, + totalTests, + totalTests ~= 0 + and 100 * (testsPassed / totalTests) + or 0) + ) + + return 0 +end, debug.traceback) + +if not result then + print("Unable to run tests due to test framework error: ",err) +end + +os.exit(result or -1) |