zig/test/cases/enum_with_members.zig

32 lines
760 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" {
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
}