enumerable-stats 1.2.1 → 1.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4b7a1951101022de006735e6276e1db4a974d98a5ae23c617a0f0c54b116ec80
-  data.tar.gz: 0efb5538568ded644e36e5f0a5ffb70cd52c86f678c490751e8c9b5987e99e46
+  metadata.gz: c0131e566d1175fe367fb75c840914ae752eea76a1a194a8d35a02d24591d6ad
+  data.tar.gz: bb99194f3965132256bb9cdabed80bf58c5da172f7cd10e4f56277c0417fe7d0
 SHA512:
-  metadata.gz: 20ddf5dd46540ff3a3ce31de0a153babcb1f005556d782e82371dbc70ddf7f882960dd4bd3197fe43571072561a812fb1ccc3b6cb8d5cb9c87cedfd61e9e1c48
-  data.tar.gz: 8bcfa97b1be3d3a1cb6887b1aa1f2ec733250361fe8a5834ef090273c97e75aebab38e0ec27d6277a843ad7ef5f8176176e41b7e39fa5d4641ad3daf319a66aa
+  metadata.gz: 022f98715f44e7788749adb509bbef27bf759d2d09c7d946ebbdf21e34ea6e8a7225e011be6ad39e3260cec3f7daad17c61ecefbd9a197457b0e75d9bd10889c
+  data.tar.gz: d1e0a0052b175fe9fc4175961d0b52248c8cc78c39258048fa7846d9e67c8bd2ff6fe6da365b41284b9b9e2d48c8a9c1fcc165d621a25d9227df88716849ae34

data/lib/enumerable_stats/enumerable_ext.rb

@@ -93,6 +93,9 @@ module EnumerableStats
     #   treatment = [15, 17, 16, 18, 14]
     #   t_stat = control.t_value(treatment)  # => ~-4.2 (negative means treatment > control)
     def t_value(other)
+      raise ArgumentError, "Cannot compare with an empty collection" if empty? || other.empty?
+      raise ArgumentError, "Parameter must be an Enumerable" unless other.respond_to?(:mean)
+
       signal = (mean - other.mean)
       noise = Math.sqrt(
         ((standard_deviation**2) / count) +
@@ -144,6 +147,16 @@ module EnumerableStats
       t_stat > critical_value
     end
 
+    # Alias for greater_than?
+    def >(other, alpha: 0.05)
+      greater_than?(other, alpha: alpha)
+    end
+
+    # Alias for less_than?
+    def <(other, alpha: 0.05)
+      less_than?(other, alpha: alpha)
+    end
+
     # Tests if this collection's mean is significantly less than another collection's mean
     # using a one-tailed Student's t-test. Returns true if the test indicates statistical
     # significance at the specified alpha level.
@@ -374,12 +387,28 @@ module EnumerableStats
         # Cauchy distribution: exact inverse
         return Math.tan(Math::PI * (0.5 - alpha))
       elsif df == 2
-        # Exact formula for df=2: t = z / sqrt(1 - z^2/(z^2 + 2))
-        # This is more numerically stable
-        z_sq = z**2
-        # Exact formula for df=2: t = z / sqrt(1 - z^2/(z^2 + 2))
-        return z / Math.sqrt(1.0 - (z_sq / (z_sq + 2.0)))
+        # Exact closed-form solution for df=2
+        # For df=2, CDF: F(t) = 1/2 * (1 + t/√(t² + 2))
+        # Quantile function: t = (2p - 1)/√(2p(1 - p)) where p = 1 - α
+
+        p = 1.0 - alpha
 
+        # Handle edge cases
+        return Float::INFINITY if p >= 1.0
+        return -Float::INFINITY if p <= 0.0
+
+        # For p very close to 0.5, use normal approximation to avoid numerical issues
+        return 0.0 if (p - 0.5).abs < 1e-10
+
+        # Apply exact formula: t = (2p - 1)/√(2p(1 - p))
+        numerator = (2.0 * p) - 1.0
+        denominator_sq = 2.0 * p * (1.0 - p)
+
+        # Ensure we don't have numerical issues with the square root
+        return numerator / Math.sqrt(denominator_sq) if denominator_sq.positive?
+
+        # Fallback to normal approximation for edge cases
+        return z
       end
 
       # Use Cornish-Fisher expansion for general case
@@ -388,29 +417,31 @@ module EnumerableStats
       # Base normal quantile
       t = z
 
-      # First-order correction
+      # First-order correction - Cornish-Fisher expansion
+      # Standard form: (z³ + z)/(4ν)
       if df >= 4
-        c1 = z / 4.0
+        c1 = ((z**3) + z) / 4.0
         t += c1 / df
       end
 
-      # Second-order correction
+      # Second-order correction - Cornish-Fisher expansion
+      # Standard form: (5z⁵ + 16z³ + 3z)/(96ν²)
       if df >= 6
-        c2 = ((5.0 * (z**3)) + (16.0 * z)) / 96.0
+        c2 = ((5.0 * (z**5)) + (16.0 * (z**3)) + (3.0 * z)) / 96.0
         t += c2 / (df**2)
       end
 
       # Third-order correction for better accuracy
+      # Standard form: (3z⁷ + 19z⁵ + 17z³ - 15z)/(384ν³)
       if df >= 8
-        c3 = ((3.0 * (z**5)) + (19.0 * (z**3)) + (17.0 * z)) / 384.0
+        c3 = ((3.0 * (z**7)) + (19.0 * (z**5)) + (17.0 * (z**3)) - (15.0 * z)) / 384.0
         t += c3 / (df**3)
       end
 
-      # Fourth-order correction for very high accuracy
-      if df >= 10
-        c4 = ((79.0 * (z**7)) + (776.0 * (z**5)) +
-          (1482.0 * (z**3)) + (776.0 * z)) / CORNISH_FISHER_FOURTH_ORDER_DENOMINATOR
-
+      # Fourth-order correction - using standard coefficients
+      # More conservative approach for high accuracy
+      if df >= 12
+        c4 = ((79.0 * (z**7)) + (776.0 * (z**5)) + (1482.0 * (z**3)) + (776.0 * z)) / CORNISH_FISHER_FOURTH_ORDER_DENOMINATOR
         t += c4 / (df**4)
       end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: enumerable-stats
 version: !ruby/object:Gem::Version
-  version: 1.2.1
+  version: 1.3.0
 platform: ruby
 authors:
 - Jon Daniel
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-08-02 00:00:00.000000000 Z
+date: 2025-08-03 00:00:00.000000000 Z
 dependencies: []
 description: |
   A Ruby gem that extends all Enumerable objects (Arrays, Ranges, Sets, etc.) with essential statistical methods.