Merge pull request #425 from VeriFIT/transition-comparison

Transition: comparison operators #patch
VeriFIT · Jul 17, 2024 · bdcb62d · bdcb62d
2 parents 8119129 + 2253cf7
commit bdcb62d
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/include/mata/nfa/delta.hh b/include/mata/nfa/delta.hh
@@ -26,7 +26,7 @@ struct Transition {
     Transition(const State source, const Symbol symbol, const State target)
             : source(source), symbol(symbol), target(target) {}
 
-    bool operator==(const Transition& rhs) const { return source == rhs.source && symbol == rhs.symbol && target == rhs.target; }
+    auto operator<=>(const Transition&) const = default;
 };
 
 /**

diff --git a/tests/nfa/delta.cc b/tests/nfa/delta.cc
@@ -467,3 +467,18 @@ TEST_CASE("mata::nfa::Delta::add_symbols_to()") {
         { "0", 0 }, { "1", 1 }, { "2", 2 }, { "3", 3 }
     });
 }
+
+TEST_CASE("Transition comparison") {
+    Transition tr1 {1, 2, 3};
+    Transition tr2 {1, 3, 1};
+    Transition tr3 {1, 2, 5};
+    Transition tr4 {1, 2, 1};
+    Transition tr5 {1, 2, 1};
+
+    CHECK(tr1 < tr2);
+    CHECK(tr2 >= tr3);
+    CHECK(tr3 >= tr1);
+    CHECK(tr4 < tr3);
+    CHECK(tr5 <= tr4);
+    CHECK(tr5 == tr4);
+}