add width() to ask width of an x86 operand

this is largely wrong for memory operands, which require more invasive changes

1 files changed, 16 insertions, 0 deletions

diff --git a/test/protected_mode/operand.rs b/test/protected_mode/operand.rs
new file mode 100644
index 0000000..08a24be
--- /dev/null
+++ b/test/protected_mode/operand.rs
@@ -0,0 +1,16 @@
+use yaxpeax_arch::{Decoder, LengthedInstruction};
+use yaxpeax_x86::long_mode::{DecodeError, InstDecoder, Opcode};
+
+#[test]
+fn register_widths() {
+    assert_eq!(Operand::Register(RegSpec::esp()).width(), 4);
+    assert_eq!(Operand::Register(RegSpec::sp()).width(), 2);
+    assert_eq!(Operand::Register(RegSpec::cl()).width(), 1);
+    assert_eq!(Operand::Register(RegSpec::ch()).width(), 1);
+    assert_eq!(Operand::Register(RegSpec::gs()).width(), 2);
+}
+
+#[test]
+fn memory_widths() {
+    assert_eq!(Operand::RegDeref(RegSpec::rsp()).width(), 4);
+}