def test_notation_i(self):
clip_z_dimension = True
m_seed = [0, 1, 2]
rmoments_order = get_m_order_as_in_r(m_seed, m_seed, m_seed)
e_seed = [0, 1, -1]
ex, ey, ez, e_new = get_e_as_in_r(e_seed, e_seed, e_seed)
if clip_z_dimension:
rmoments_order = rmoments_order[0:9]
q, d = rmoments_order.shape
d = 2
ex = ex[0:9]
ey = ey[0:9]
ez = ez[0:9]
e_D2Q9 = e_new[0:9, :]
else:
q, d = rmoments_order.shape
expected_e = Matrix([[0, 0, 0], [1, 0, 0], [-1, 0, 0], [0, 1, 0],
[1, 1, 0], [-1, 1, 0], [0, -1, 0], [1, -1, 0],
[-1, -1, 0]])
assert e_D2Q9 == expected_e
rev_i = get_reverse_indices(e_D2Q9)
expected_rev_i = np.array([0, 2, 1, 6, 8, 7, 3, 5, 4])
np.testing.assert_array_equal(rev_i, expected_rev_i)
import numpy as np
import pandas as pd
from SymbolicCollisions.core.cm_symbols import moments_dict, e_D3Q27, ex_D3Q27
from SymbolicCollisions.core.MatrixGenerator import get_m_order_as_in_r, get_e_as_in_r
e_seed = [0, 1, -1]
grid = get_m_order_as_in_r(e_seed, e_seed, e_seed)
moments_order = grid.to_numpy()
print(f"order of moments: \n {moments_order}")
ex_D3Q27new, ey_D3Q27new, ez_D3Q27new, e_D3Q27new = get_e_as_in_r(
e_seed, e_seed, e_seed)
print(f"lattice velocities - e: \n {e_D3Q27new}")
print(e_D3Q27)
print(ex_D3Q27)
print(ex_D3Q27new)
print()
# > U
# Var1 Var2 Var3
# 1 0 0 0
# 2 1 0 0
# 3 -1 0 0
# 4 0 1 0
# 5 1 1 0
# 6 -1 1 0
# 7 0 -1 0
# 8 1 -1 0
# 9 -1 -1 0
# 10 0 0 1
import pandas as pd
from sympy import Symbol
import time
start = time.process_time()
########################################
clip_z_dimension = True
m_seed = [0, 1, 2]
rmoments_order = get_m_order_as_in_r(m_seed, m_seed, m_seed)
e_seed = [0, 1, -1]
ex, ey, ez, e_new = get_e_as_in_r(e_seed, e_seed, e_seed)
if clip_z_dimension:
rmoments_order = rmoments_order[0:9]
q, d = rmoments_order.shape
d = 2
ex = ex[0:9]
ey = ey[0:9]
ez = ez[0:9]
e_D2Q9 = e_new[0:9, :]
else:
q, d = rmoments_order.shape
print(f"order of rmoments: \n {pd.DataFrame.from_records(rmoments_order)}")
print(f"lattice velocities - e: \n {np.array(e_D2Q9)}")
