From b0ad6748e428d5a4697d6fd78510e173c16becf3 Mon Sep 17 00:00:00 2001
From: sanket1729 <sanket1729@gmail.com>
Date: Mon, 13 Sep 2021 07:45:15 -0700
Subject: [PATCH] Add tests for opcode classification

---
 src/blockdata/opcodes.rs | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/src/blockdata/opcodes.rs b/src/blockdata/opcodes.rs
index 57e7063e..62c13a3e 100644
--- a/src/blockdata/opcodes.rs
+++ b/src/blockdata/opcodes.rs
@@ -888,6 +888,17 @@ mod tests {
         }
     }
 
+    #[test]
+    fn classify_test() {
+        let op186 = all::OP_CHECKSIGADD;
+        assert_eq!(op186.classify(ClassifyContext::Legacy), Class::ReturnOp);
+        assert_eq!(op186.classify(ClassifyContext::TapScript), Class::Ordinary(Ordinary::OP_CHECKSIGADD));
+
+        let op187 = all::OP_RETURN_187;
+        assert_eq!(op187.classify(ClassifyContext::Legacy), Class::ReturnOp);
+        assert_eq!(op187.classify(ClassifyContext::TapScript), Class::SuccessOp);
+    }
+
     #[test]
     fn str_roundtrip() {
         let mut unique = HashSet::new();