24 lines
527 B
Python
24 lines
527 B
Python
|
import gaspype as gp
|
||
|
|
import numpy as np
|
||
|
|
import pandas as pd
|
||
|
|
|
||
|
|
|
||
|
|
def test_fluid():
|
||
|
|
fl = gp.fluid({'O2': 1, 'H2': 2, 'H2O': 3})
|
||
|
|
|
||
|
|
df = pd.DataFrame(list(fl))
|
||
|
|
assert df.shape == (1, 3)
|
||
|
|
|
||
|
|
df = pd.DataFrame(list(fl * np.array([1, 2, 3, 4])))
|
||
|
|
assert df.shape == (4, 3)
|
||
|
|
|
||
|
|
|
||
|
|
def test_elements():
|
||
|
|
fl = gp.fluid({'O2': 1, 'H2': 2, 'H2O': 3})
|
||
|
|
|
||
|
|
df = pd.DataFrame(list(gp.elements(fl)))
|
||
|
|
assert df.shape == (1, 2)
|
||
|
|
|
||
|
|
df = pd.DataFrame(list(gp.elements(fl * np.array([1, 2, 3, 4]))))
|
||
|
|
assert df.shape == (4, 2)
|