Trio/TrioTests/OpenAPSSwiftTests/utils/Extensions.swift

import Foundation
@testable import Trio

extension [ComputedPumpHistoryEvent] {
    func netInsulin() -> Decimal { compactMap(\.insulin).reduce(0, +) }
}

extension Decimal {
    func isWithin(_ error: Decimal, of value: Decimal) -> Bool {
        (self - value).magnitude <= error
    }
}

extension Encodable {
    var prettyPrintedJSON: String? {
        let encoder = JSONCoding.encoder
        encoder.outputFormatting = [.prettyPrinted, .withoutEscapingSlashes, .sortedKeys]

        do {
            let data = try encoder.encode(self)
            return String(data: data, encoding: .utf8)
        } catch {
            return nil
        }
    }
}

extension IobResult {
    func approximatelyEquals(_ rhs: IobResult) -> Bool {
        // Compare all properties
        guard iob.isWithin(0.001, of: rhs.iob),
              activity.isWithin(0.0001, of: rhs.activity),
              basaliob.isWithin(0.001, of: rhs.basaliob),
              bolusiob.isWithin(0.001, of: rhs.bolusiob),
              netbasalinsulin.isWithin(0.05, of: rhs.netbasalinsulin),
              bolusinsulin.isWithin(0.001, of: rhs.bolusinsulin),
              time == rhs.time,
              lastBolusTime == rhs.lastBolusTime
        else {
            return false
        }

        // Compare nested IobWithZeroTemp
        guard iobWithZeroTemp.iob.isWithin(0.001, of: rhs.iobWithZeroTemp.iob),
              iobWithZeroTemp.activity.isWithin(0.0001, of: rhs.iobWithZeroTemp.activity),
              iobWithZeroTemp.basaliob.isWithin(0.001, of: rhs.iobWithZeroTemp.basaliob),
              iobWithZeroTemp.bolusiob.isWithin(0.001, of: rhs.iobWithZeroTemp.bolusiob),
              iobWithZeroTemp.netbasalinsulin.isWithin(0.05, of: rhs.iobWithZeroTemp.netbasalinsulin),
              iobWithZeroTemp.bolusinsulin.isWithin(0.001, of: rhs.iobWithZeroTemp.bolusinsulin),
              iobWithZeroTemp.time == rhs.iobWithZeroTemp.time
        else {
            return false
        }

        // Compare optional LastTemp
        if let selfTemp = lastTemp, let rhsTemp = rhs.lastTemp {
            guard let selfDuration = selfTemp.duration, let rhsDuration = rhsTemp.duration, selfDuration.isWithin(
                0.01,
                of: rhsDuration
            ) else {
                return false
            }
            // Both are non-nil, compare their properties
            return selfTemp.rate == rhsTemp.rate &&
                selfTemp.timestamp == rhsTemp.timestamp &&
                selfTemp.started_at == rhsTemp.started_at &&
                selfTemp.date == rhsTemp.date
        } else {
            // Both should be nil for equality
            return lastTemp == nil && rhs.lastTemp == nil
        }
    }
}