feat(med) - Add support to F64 for median operation

Author: protocolstardust

core/math.c

@@ -2650,11 +2650,10 @@ obj_p ray_med(obj_p x) {
     if (l == 0)
         return f64(NULL_F64);
 
-    // i32_t *xi32sort;
     i64_t *xisort;
     u8_t *xu8sort;
     i16_t *xi16sort;
-    // f64_t *xfsort, med;
+    f64_t *xfsort;
     f64_t med;
     obj_p sort;
 
@@ -2700,14 +2699,13 @@ obj_p ray_med(obj_p x) {
             drop_obj(sort);
 
             return f64(med);
-            // TODO
-            // case TYPE_F64:
-            //     sort = ray_asc(x);
-            //     xfsort = AS_F64(sort);
-            //     med = (l % 2 == 0) ? (xfsort[l / 2 - 1] + xfsort[l / 2]) / 2.0 : xfsort[l / 2];
-            //     drop_obj(sort);
 
-            //     return f64(med);
+        case TYPE_F64:
+            sort = ray_asc(x);
+            xfsort = AS_F64(sort);
+            med = (l % 2 == 0) ? (xfsort[l / 2 - 1] + xfsort[l / 2]) / 2.0 : xfsort[l / 2];
+            drop_obj(sort);
+            return f64(med);
 
         case TYPE_PARTEDI16:
         case TYPE_PARTEDI32:

tests/aggr_tests.c

@@ -283,6 +283,31 @@ test_result_t test_aggr_median() {
     // All same values
     TEST_ASSERT_EQ("(med [5 5 5 5 5])", "5.0");
 
+    // F64 odd length
+    TEST_ASSERT_EQ("(med [1.0 2.0 3.0 4.0 5.0])", "3.0");
+    TEST_ASSERT_EQ("(med [5.0 1.0 3.0 2.0 4.0])", "3.0");
+    TEST_ASSERT_EQ("(med [100.5 200.5 300.5])", "200.5");
+
+    // F64 even length
+    TEST_ASSERT_EQ("(med [1.0 2.0 3.0 4.0])", "2.5");
+    TEST_ASSERT_EQ("(med [10.0 20.0 30.0 40.0])", "25.0");
+
+    // F64 single element
+    TEST_ASSERT_EQ("(med [42.5])", "42.5");
+
+    // F64 two elements
+    TEST_ASSERT_EQ("(med [1.5 3.5])", "2.5");
+
+    // F64 negative values
+    TEST_ASSERT_EQ("(med [-5.0 -3.0 -1.0 1.0 3.0])", "-1.0");
+    TEST_ASSERT_EQ("(med [-10.0 0.0 10.0])", "0.0");
+
+    // F64 all same values
+    TEST_ASSERT_EQ("(med [5.5 5.5 5.5 5.5 5.5])", "5.5");
+
+    // F64 scalar
+    TEST_ASSERT_EQ("(med 3.14)", "3.14");
+
     PASS();
 }