docs: def calc_partition_function(mass, omega_array, temperature_array):
      """
      Calculates the partition function of your system at each point in time.
  
      Parameters
      ----------
      mass : float
          The mass of the particle in kg
      omega_array : array
          array which represents omega at every point in your time trace
          and should therefore have the same length as the Hamiltonian
      temperature_array : array
          array which represents the temperature at every point in your time trace
          and should therefore have the same length as the Hamiltonian
  
      Returns:
      -------
      Partition function : array
          The Partition Function at every point in time over a given trap-frequency and temperature change.
      """
      Kappa_t= mass*omega_array**2    
      return _np.sqrt(4*_np.pi**2*_scipy.constants.Boltzmann**2*temperature_array**2/(mass*Kappa_t))