93 lines
2.0 KiB
Lua
93 lines
2.0 KiB
Lua
utils = require('utils')
|
|
|
|
function with_temp_ref(...)
|
|
local args = {...}
|
|
local dtype = 'int32_t'
|
|
if type(args[1]) == 'string' then
|
|
dtype = table.remove(args, 1)
|
|
end
|
|
local count = 1
|
|
if type(args[1]) == 'number' then
|
|
count = table.remove(args, 1)
|
|
end
|
|
local func = table.remove(args, 1)
|
|
assert(type(func) == 'function')
|
|
dfhack.with_temp_object(df.new(dtype, count), func, table.unpack(args))
|
|
end
|
|
|
|
function test.zero_initialize()
|
|
with_temp_ref(function(x)
|
|
expect.eq(x.value, 0)
|
|
end)
|
|
end
|
|
|
|
function test.zero_initialize_array()
|
|
with_temp_ref(2, function(x)
|
|
expect.eq(x:_displace(1).value, 0)
|
|
end)
|
|
end
|
|
|
|
function test.displace_zero()
|
|
with_temp_ref(function(x)
|
|
expect.eq(x, x:_displace(0))
|
|
end)
|
|
end
|
|
|
|
function test.displace_nonzero()
|
|
with_temp_ref(2, function(x)
|
|
expect.ne(x, x:_displace(1))
|
|
expect.eq(utils.addressof(x) + x:sizeof(), utils.addressof(x:_displace(1)))
|
|
end)
|
|
end
|
|
|
|
function test.displace_negative()
|
|
with_temp_ref(2, function(x)
|
|
expect.true_(x:_displace(1):_displace(-1).value)
|
|
end)
|
|
end
|
|
|
|
function test.index_read()
|
|
with_temp_ref(function(x)
|
|
expect.eq(x.value, x[0])
|
|
end)
|
|
end
|
|
|
|
function test.index_write()
|
|
with_temp_ref(function(x)
|
|
x[0] = 1
|
|
expect.eq(x.value, 1)
|
|
expect.eq(x[0], 1)
|
|
x.value = 2
|
|
expect.eq(x.value, 2)
|
|
expect.eq(x[0], 2)
|
|
end)
|
|
end
|
|
|
|
function test.index_write_multi()
|
|
local len = 3
|
|
with_temp_ref(len, function(x)
|
|
for i = 0, len - 1 do
|
|
x[i] = i * i
|
|
end
|
|
for i = 0, len - 1 do
|
|
expect.eq(x[i], i * i, i)
|
|
end
|
|
end)
|
|
end
|
|
|
|
function test.index_read_negative()
|
|
with_temp_ref(function(x)
|
|
expect.error_match('negative index', function()
|
|
expect.true_(x:_displace(1)[-1])
|
|
end)
|
|
end)
|
|
end
|
|
|
|
function test.index_write_negative()
|
|
with_temp_ref(function(x)
|
|
expect.error_match('negative index', function()
|
|
x:_displace(1)[-1] = 7
|
|
end)
|
|
end)
|
|
end
|