zig/test/cases/enum_with_members.zig

28 lines
732 B
Zig
Raw Normal View History

2017-01-05 00:57:48 -08:00
const assert = @import("std").debug.assert;
const mem = @import("std").mem;
const fmt = @import("std").fmt;
2017-01-05 00:57:48 -08:00
const ET = union(enum) {
2016-12-25 22:37:33 -08:00
SINT: i32,
UINT: u32,
pub fn print(a: *const ET, buf: []u8) error!usize {
return switch (a.*) {
ET.SINT => |x| fmt.formatIntBuf(buf, x, 10, false, 0),
ET.UINT => |x| fmt.formatIntBuf(buf, x, 10, false, 0),
};
2016-12-25 22:37:33 -08:00
}
};
2017-05-23 18:38:31 -07:00
test "enum with members" {
2018-05-28 17:23:55 -07:00
const a = ET{ .SINT = -42 };
const b = ET{ .UINT = 42 };
2016-12-25 22:37:33 -08:00
var buf: [20]u8 = undefined;
assert((a.print(buf[0..]) catch unreachable) == 3);
assert(mem.eql(u8, buf[0..3], "-42"));
2016-12-25 22:37:33 -08:00
assert((b.print(buf[0..]) catch unreachable) == 2);
assert(mem.eql(u8, buf[0..2], "42"));
2016-12-25 22:37:33 -08:00
}