def test_calculate(): calc = calculate_flux(XYZ) assert isinstance(calc, list) assert len(calc) == 11 assert len(calc[0]) == 5 *orig, dol, perc = calc[0] assert orig == ["Cash", 120000, 115000] assert dol == 5000 assert round(perc, 2) == 0.04
def test_identify(): flux = identify_flux(calculate_flux(XYZ)) assert isinstance(flux, list) assert len(flux) == 5 assert [act for act, *_ in flux] == [ "Accounts Receivable", "Inventory", "Notes Receivable", "Accrued Payroll", "Retained Earnings", ]
if save_fig: basis.save_plot(save_location, iteration, time, x, state, sol, n_ex, plot_limits) iteration += 1 while time < end_time: # Calculate time step dt = CLF*dx/basis.time_step(state) if time + dt >= end_time: dt = end_time - time print(time) # Time integration flux1 = -dt*flux.calculate_flux(state, flux_p, flux_n, phi, n_ex,fix_cavitation)/dx temp = state + flux1 if fix_cavitation: temp = basis.fix_cavitation(temp) flux2 = -dt*flux.calculate_flux(temp, flux_p, flux_n, phi, n_ex,fix_cavitation)/dx temp = state + .25*flux1 + .25*flux2 if fix_cavitation: temp = basis.fix_cavitation(temp) flux3 = -dt*flux.calculate_flux(temp, flux_p, flux_n, phi, n_ex,fix_cavitation)/dx state = state + (flux1 + flux2 + 4*flux3)/6 if fix_cavitation: state = basis.fix_cavitation(state)