zig/test/behavior/enum_with_members.zig

28 lines
789 B
Zig
Raw Normal View History

const expect = @import("std").testing.expect;
const mem = @import("std").mem;
const fmt = @import("std").fmt;
2017-01-05 08:57:48 +00:00
const ET = union(enum) {
2016-12-26 06:37:33 +00:00
SINT: i32,
UINT: u32,
pub fn print(a: *const ET, buf: []u8) anyerror!usize {
return switch (a.*) {
ET.SINT => |x| fmt.formatIntBuf(buf, x, 10, false, fmt.FormatOptions{}),
ET.UINT => |x| fmt.formatIntBuf(buf, x, 10, false, fmt.FormatOptions{}),
};
2016-12-26 06:37:33 +00:00
}
};
2017-05-24 02:38:31 +01:00
test "enum with members" {
const a = ET{ .SINT = -42 };
const b = ET{ .UINT = 42 };
2016-12-26 06:37:33 +00:00
var buf: [20]u8 = undefined;
try expect((a.print(buf[0..]) catch unreachable) == 3);
try expect(mem.eql(u8, buf[0..3], "-42"));
2016-12-26 06:37:33 +00:00
try expect((b.print(buf[0..]) catch unreachable) == 2);
try expect(mem.eql(u8, buf[0..2], "42"));
2016-12-26 06:37:33 +00:00
}