From 436f9849503ab838fe2cc59a25ae6517aafec408 Mon Sep 17 00:00:00 2001
From: Jeong Yunwon <jeong@youknowone.org>
Date: Thu, 12 May 2022 04:17:25 +0900
Subject: [PATCH] clean up PyComparisonOp::map_eq

---
 vm/src/types/slot.rs | 25 +++++++++----------------
 1 file changed, 9 insertions(+), 16 deletions(-)

diff --git a/vm/src/types/slot.rs b/vm/src/types/slot.rs
index 3a74086d7..5d29231be 100644
--- a/vm/src/types/slot.rs
+++ b/vm/src/types/slot.rs
@@ -776,22 +776,15 @@ impl PyComparisonOp {
     /// is `Ne` and `f()` returns true. Otherwise returns `None`.
     #[inline]
     pub fn map_eq(self, f: impl FnOnce() -> bool) -> Option<bool> {
-        match self {
-            Self::Eq => {
-                if f() {
-                    Some(true)
-                } else {
-                    None
-                }
-            }
-            Self::Ne => {
-                if f() {
-                    Some(false)
-                } else {
-                    None
-                }
-            }
-            _ => None,
+        let eq = match self {
+            Self::Eq => true,
+            Self::Ne => false,
+            _ => return None,
+        };
+        if f() {
+            Some(eq)
+        } else {
+            None
         }
     }
 }