Add support for <=> to fn\operator()

nikic · Jul 3, 2015 · 7da440d · 7da440d
1 parent e9c2ab7
commit 7da440d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/iter.fn.php b/src/iter.fn.php
@@ -90,6 +90,9 @@ function operator($operator, $arg = null) {
         '&&'  => function($a, $b) { return $a &&  $b; },
         '||'  => function($a, $b) { return $a ||  $b; },
         '**'  => function($a, $b) { return \pow($a, $b); },
+        '<=>' => function($a, $b) {
+            return $a == $b ? 0 : ($a < $b ? -1 : 1);
+        },
     ];
 
     if (!isset($functions[$operator])) {

diff --git a/test/IterFnTest.php b/test/IterFnTest.php
@@ -102,6 +102,10 @@ public function provideTestOperator() {
             ['&&', true, false, false],
             ['||', true, false, true],
             ['**', 2, 4, 16],
+            ['<=>', [0=>1,1=>0], [1=>0,0=>1], 0],
+            ['<=>', '2e1', '1e10', -1],
+            ['<=>', new \stdClass(), new \SplStack(), 1],
+            ['<=>', new \SplStack(), new \stdClass(), 1],
         ];
     }