50 lines
1.2 KiB
Python
50 lines
1.2 KiB
Python
|
import gaspype as gp
|
||
|
|
|
||
|
|
|
||
|
|
def test_set_and_read_solver():
|
||
|
|
tmp = gp.get_solver()
|
||
|
|
gp.set_solver('gibs minimization')
|
||
|
|
assert gp.get_solver() == 'gibs minimization'
|
||
|
|
gp.set_solver(tmp)
|
||
|
|
|
||
|
|
|
||
|
|
def test_fluid_stacking_concat():
|
||
|
|
fl1 = gp.fluid({'O2': 1, 'N2': 0})
|
||
|
|
fl2 = gp.fluid({'N2': 1}, fl1.fs)
|
||
|
|
|
||
|
|
fl3 = gp.stack([fl1, fl2])
|
||
|
|
assert fl3.shape == (2,)
|
||
|
|
|
||
|
|
fl4 = gp.stack([fl3, fl3])
|
||
|
|
assert fl4.shape == (2, 2)
|
||
|
|
|
||
|
|
fl5 = gp.concat([fl3, fl3])
|
||
|
|
assert fl5.shape == (4,)
|
||
|
|
|
||
|
|
|
||
|
|
def test_elements_stacking_concat():
|
||
|
|
el1 = gp.elements(gp.fluid({'O2': 1, 'N2': 0}))
|
||
|
|
el2 = gp.elements(gp.fluid({'N2': 1}), el1.fs)
|
||
|
|
|
||
|
|
assert el1.fs == el2.fs
|
||
|
|
|
||
|
|
el3 = gp.stack([el1, el2])
|
||
|
|
assert el3.shape == (2,)
|
||
|
|
|
||
|
|
el4 = gp.stack([el3, el3])
|
||
|
|
assert el4.shape == (2, 2)
|
||
|
|
|
||
|
|
el5 = gp.concat([el3, el3])
|
||
|
|
assert el5.shape == (4,)
|
||
|
|
|
||
|
|
|
||
|
|
def test_element_casting():
|
||
|
|
fl1 = gp.fluid({'O2': 1, 'N2': 2, 'H2': 3})
|
||
|
|
|
||
|
|
el1 = gp.elements(fl1)
|
||
|
|
assert el1.get_elemental_composition() == {'N': 4.0, 'H': 6.0, 'O': 2.0}
|
||
|
|
|
||
|
|
fs = gp.fluid_system('H2, O2, N2')
|
||
|
|
el2 = gp.elements(fl1, fs)
|
||
|
|
assert el2.get_elemental_composition() == {'N': 4.0, 'H': 6.0, 'O': 2.0}
|