def test_get_indexed_variables(mass, grid):
""" Ensure that all of the Indexed grid variables are returned. """
variables, count = get_indexed_variables([mass.expanded])
rho = IndexedBase("rho")
rhou0 = IndexedBase("rhou0")
rhou1 = IndexedBase("rhou1")
x0 = EinsteinTerm("x0")
x1 = EinsteinTerm("x1")
t = EinsteinTerm("t")
assert variables == [rho[x0, x1, t], rhou0[x0, x1, t], rhou1[x0, x1, t]]
return
评论列表
文章目录