Wavefunction Probability Density Calculator

            from math import *

def wavefunction_probability(a, b) -> float:
    return ((a * a) + (b * b))

assert abs(wavefunction_probability(0.6, 0.8) - 1) < 1e-6 * max(1.0, abs(1))
          

            #include <assert.h>
#include <math.h>

double wavefunction_probability(double a, double b) {
    return ((a * a) + (b * b));
}

int main(void) {
    const double expected = 1;
    const double actual = wavefunction_probability(0.6, 0.8);
    assert(fabs(actual - expected) < 1e-6 * fmax(1.0, fabs(expected)));
}
          

            #include <cassert>
#include <cmath>
#include <numbers>

double wavefunction_probability(double a, double b) {
    return ((a * a) + (b * b));
}

int main() {
    constexpr double expected = 1;
    const double actual = wavefunction_probability(0.6, 0.8);
    assert(std::fabs(actual - expected) < 1e-6 * std::fmax(1.0, std::fabs(expected)));
}
          

x86-64 NASM · System V ABI · Linux · SSE2 with libm where required

            ; double wavefunction_probability(double a, double b)
; Linux x86-64 NASM · System V ABI · first eight doubles in xmm0–xmm7
global wavefunction_probability
section .text

wavefunction_probability:
    push rbp
    mov rbp, rsp
    sub rsp, 48
    movsd [rbp-8], xmm0
    movsd [rbp-16], xmm1
    movsd xmm0, [rbp-8]
    mulsd xmm0, [rbp-8]
    movsd [rbp-32], xmm0
    movsd xmm0, [rbp-16]
    mulsd xmm0, [rbp-16]
    movsd [rbp-40], xmm0
    movsd xmm0, [rbp-32]
    addsd xmm0, [rbp-40]
    movsd [rbp-24], xmm0
    movsd xmm0, [rbp-24]
    leave
    ret
          

            function result = wavefunction_probability(a, b)
    result = ((a * a) + (b * b));
end
          

            ClearAll[mwCalculate];
mwCalculate[a_, b_] := ((a * a) + (b * b));