def test_dive_plan(self):
gas = buhlmann.Gas(n2_pc=0.79, he_pc=0.0)
initial_tissues = buhlmann.Tissues()
dive_plan = pd.DataFrame([[0, 0], [2, 40], [22, 40]],
columns=['t', 'depth'])
run_dive_plan = buhlmann.run_dive(dive_plan, initial_tissues, gas)
current_tissues = run_dive_plan.iloc[-1]['tissues']
current_depth = run_dive_plan.iloc[-1]['depth']
max_ascent_rate = 9 # metres/sec
run_stops = buhlmann.get_stops_to_surface(current_tissues,
current_depth, gas,
max_ascent_rate)
run_stops['t'] += run_dive_plan.iloc[-1]['t']
run_dive_plan = run_dive_plan.append(run_stops.iloc[1:])
run_dive_plan.reset_index(inplace=True, drop=True)
# get data resolution at 1 minute
dive_data = buhlmann.run_dive(run_dive_plan,
initial_tissues,
gas,
resolution=1)
# check that we have all times in minutes
self.assertTrue(dive_data.iloc[-1]['t'] > dive_plan.iloc[-1]['t'], )
# check that every time is there
self.assertEqual(list(dive_data['t']), list(range(len(dive_data))))
# check that we surfaced
self.assertEqual(dive_data.iloc[-1]['depth'], 0)
def test_zero_time(self):
gas = buhlmann.Gas(n2_pc=0.45, he_pc=0.40)
tissue = buhlmann.Tissues()
new_tissue = buhlmann.get_partial_pressures(
tissue, gas, buhlmann.depth_to_pressure(0),
buhlmann.depth_to_pressure(40), 0)
npt.assert_array_equal(tissue.n2_p, new_tissue.n2_p)
npt.assert_array_equal(tissue.he_p, new_tissue.he_p)
def test_flat_bottom(self):
gas = buhlmann.Gas(n2_pc=0.79, he_pc=0.0)
tissue = buhlmann.Tissues()
new_tissue = buhlmann.get_partial_pressures(
tissue, gas, buhlmann.depth_to_pressure(40),
buhlmann.depth_to_pressure(40), 20)
self.assertEqual(
math.ceil(
buhlmann.pressure_to_depth(
buhlmann.ceiling_pressure(new_tissue))), 6)
def test_gradient_factors(self):
initial_tissues = buhlmann.Tissues()
for gf_pc in list(range(0, 110, 10)):
print("GF={}, ceiling={}".format(
gf_pc / 100.0,
buhlmann.pressure_to_depth(
buhlmann.ceiling_pressure(initial_tissues,
gf=gf_pc / 100.0))))
gas = buhlmann.Gas(n2_pc=0.79, he_pc=0.0)
initial_tissues = buhlmann.Tissues()
dive_plan = pd.DataFrame([[0, 0], [2, 40], [22, 40]],
columns=['t', 'depth'])
run_dive_plan = buhlmann.run_dive(dive_plan, initial_tissues, gas)
last_tissues = run_dive_plan.iloc[-1]['tissues']
for gf_pc in list(range(0, 110, 10)):
print("GF={}, ceiling={}".format(
gf_pc / 100.0,
buhlmann.pressure_to_depth(
buhlmann.ceiling_pressure(last_tissues,
gf=gf_pc / 100.0))))
def test_gradient_factors_stops(self):
gas = buhlmann.Gas(n2_pc=0.79, he_pc=0.0)
initial_tissues = buhlmann.Tissues()
dive_plan = pd.DataFrame([[0, 0], [2, 40], [22, 40]],
columns=['t', 'depth'])
gf_lo = 0.3
gf_hi = 0.8
run_dive_plan = buhlmann.run_dive(dive_plan, initial_tissues, gas)
current_tissues = run_dive_plan.iloc[-1]['tissues']
current_depth = run_dive_plan.iloc[-1]['depth']
max_ascent_rate = 9 # metres/sec
run_stops = buhlmann.get_stops_to_surface(
current_tissues,
current_depth,
gas,
max_ascent_rate,
gf_lo=gf_lo,
gf_hi=gf_hi,
)
self.assertTrue(
run_stops['gf'].between(gf_lo, gf_hi).all(),
"All GF values should be between gf_lo={} and gf_hi={}".format(
gf_lo, gf_hi))
depth_first_stop = run_dive_plan.iloc[-1]['depth']
t_first_stop = run_dive_plan.iloc[-1]['t']
run_stops['t'] += t_first_stop
run_dive_plan = run_dive_plan.append(run_stops)
# print(run_dive_plan)
dive_data = buhlmann.run_dive(
run_dive_plan,
initial_tissues,
gas,
resolution=1,
gf=buhlmann.GradientFactors(
gf_lo, buhlmann.depth_to_pressure(depth_first_stop), gf_hi,
buhlmann.depth_to_pressure(0), t_first_stop))