def test_heat_coef_drilling(self):
from pwptemp.drilling import input, heatcoefficients, initcond
depths = wellpath.get(100, 50)
well = input.set_well(input.data(), depths)
ic = initcond.init_cond(well)
well = well.define_viscosity(ic)
well = well.define_density(ic)
hc = heatcoefficients.heat_coef(well, 1)
self.assertEqual(len(hc), 9)
self.assertEqual(len(hc[0]), 4) # len of hc_1
self.assertEqual(len(hc[0][0]), depths.zstep) # len of c1z
self.assertEqual(len(hc[0][1]), depths.zstep) # len of c1e
self.assertEqual(len(hc[0][2]), depths.zstep) # len of c1
self.assertEqual(len(hc[0][3]), depths.zstep) # len of c1t
self.assertEqual(len(hc[1]), 4) # len of hc_2
self.assertEqual(len(hc[1][0]), depths.zstep) # len of c2z
self.assertEqual(len(hc[1][1]), depths.zstep) # len of c2e
self.assertEqual(len(hc[1][2]), depths.zstep) # len of c2w
self.assertEqual(len(hc[1][3]), depths.zstep) # len of c2t
self.assertEqual(len(hc[2]), 5) # len of hc_3
self.assertEqual(len(hc[2][0]), depths.zstep) # len of c3z
self.assertEqual(len(hc[2][1]), depths.zstep) # len of c3e
self.assertEqual(len(hc[2][2]), depths.zstep) # len of c3w
self.assertEqual(len(hc[2][3]), depths.zstep) # len of c3
self.assertEqual(len(hc[2][4]), depths.zstep) # len of c3t
self.assertEqual(len(hc[3]), 4) # len of hc_4
self.assertEqual(len(hc[3][0]), depths.zstep) # len of c4z
self.assertEqual(len(hc[3][1]), depths.zstep) # len of c4e
self.assertEqual(len(hc[3][2]), depths.zstep) # len of c4w
self.assertEqual(len(hc[3][3]), depths.zstep) # len of c4t
self.assertEqual(len(hc[4]), 4) # len of hc_5
self.assertEqual(len(hc[4][0]), depths.zstep) # len of c5z
self.assertEqual(len(hc[4][1]), depths.zstep) # len of c5e
self.assertEqual(len(hc[4][2]), depths.zstep) # len of c5w
self.assertEqual(len(hc[4][3]), depths.zstep) # len of c5t
def test_heat_coef_injection(self):
from pwptemp.injection import input, heatcoefficients, initcond
depths = wellpath.get(100, 50)
well = input.set_well(input.data(), depths)
ic = initcond.init_cond(well)
well = well.define_density(ic)
tt = ic.tto
t3 = ic.tco
hc = heatcoefficients.heat_coef(well, 1, tt, t3)
self.assertEqual(len(hc), 8)
self.assertEqual(len(hc[0]), 4) # len of hc_1
self.assertEqual(len(hc[0][0]), depths.zstep) # len of c1z
self.assertEqual(len(hc[0][1]), depths.zstep) # len of c1e
self.assertEqual(len(hc[0][2]), depths.zstep) # len of c1
self.assertEqual(len(hc[0][3]), depths.zstep) # len of c1t
self.assertEqual(len(hc[1]), 4) # len of hc_2
self.assertEqual(len(hc[1][0]), depths.zstep) # len of c2z
self.assertEqual(len(hc[1][1]), depths.zstep) # len of c2e
self.assertEqual(len(hc[1][2]), depths.zstep) # len of c2w
self.assertEqual(len(hc[1][3]), depths.zstep) # len of c2t
self.assertEqual(len(hc[2]), 4) # len of hc_3
self.assertEqual(len(hc[2][0]), depths.zstep) # len of c3z
self.assertEqual(len(hc[2][1]), depths.zstep) # len of c3e
self.assertEqual(len(hc[2][2]), depths.zstep) # len of c3w
self.assertEqual(len(hc[2][3]), depths.zstep) # len of c3t
self.assertEqual(len(hc[3]), 4) # len of hc_4
self.assertEqual(len(hc[3][0]), depths.zstep) # len of c4z
self.assertEqual(len(hc[3][1]), depths.zstep) # len of c4e
self.assertEqual(len(hc[3][2]), depths.zstep) # len of c4w
self.assertEqual(len(hc[3][3]), depths.zstep) # len of c4t
self.assertEqual(len(hc[4]), 4) # len of hc_5
self.assertEqual(len(hc[4][0]), depths.zstep) # len of c5z
self.assertEqual(len(hc[4][1]), depths.zstep) # len of c5e
self.assertEqual(len(hc[4][2]), depths.zstep) # len of c5w
self.assertEqual(len(hc[4][3]), depths.zstep) # len of c5t
def test_temp_calc(self):
tdata = input.data()
depths = wellpath.get(3000, 50)
well = input.set_well(tdata, depths)
ic = initcond.init_cond(well)
hc = heatcoefficients.heat_coef(well, 1)
tdist = linearsystem.temp_calc(well, ic, hc)
self.assertEqual(len(tdist.tdsi), len(tdist.ta), len(tdist.tr))
self.assertEqual(len(tdist.tr), len(tdist.tcsg), len(tdist.tsr))
def test_temp_calc_drilling(self):
from pwptemp.drilling import input, initcond, heatcoefficients, linearsystem
tdata = input.data()
depths = wellpath.get(3000, 50)
well = input.set_well(tdata, depths)
ic = initcond.init_cond(well)
well = well.define_viscosity(ic)
well = well.define_density(ic, cond=0)
hc = heatcoefficients.heat_coef(well, 1)
tdist = linearsystem.temp_calc(well, ic, hc)
self.assertEqual(len(tdist.tdsi), len(tdist.ta), len(tdist.tr))
self.assertEqual(len(tdist.tr), len(tdist.tcsg), len(tdist.tsr))
def test_temp_calc_injection(self):
from pwptemp.injection import input, initcond, heatcoefficients, linearsystem
tdata = input.data()
depths = wellpath.get(3000, 50)
well = input.set_well(tdata, depths)
ic = initcond.init_cond(well)
tt = ic.tto
tc = ic.tco
well = well.define_density(ic, cond=0)
hc = heatcoefficients.heat_coef(well, 1, tt, tc)
tdist = linearsystem.temp_calc(well, ic, hc)
self.assertEqual(len(tdist.tft), len(tdist.ta), len(tdist.tr))
self.assertEqual(len(tdist.tr), len(tdist.tc), len(tdist.tsr))
def test_init_cond(self):
tdata = input.data()
depths = wellpath.get(100, 10)
well = input.set_well(tdata, depths)
ic = initcond.init_cond(well)
a = ic.tdsio
b = ic.tdso
c = ic.tao
d = ic.tcsgo
e = ic.tsro
f = ic.tfm
self.assertIsInstance(a, list)
self.assertIsInstance(b, list)
self.assertIsInstance(c, list)
self.assertIsInstance(d, list)
self.assertIsInstance(e, list)
self.assertIsInstance(f, list)
def test_init_cond_injection(self):
from pwptemp.injection import input, initcond
tdata = input.data()
depths = wellpath.get(100, 10)
well = input.set_well(tdata, depths)
ic = initcond.init_cond(well)
a = ic.tfto
b = ic.tto
c = ic.tao
d = ic.tco
e = ic.tsro
f = ic.tfm
self.assertIsInstance(a, list)
self.assertIsInstance(b, list)
self.assertIsInstance(c, list)
self.assertIsInstance(d, list)
self.assertIsInstance(e, list)
self.assertIsInstance(f, list)
