diff options
Diffstat (limited to 'test_polymatrix/test_tomatrixrepr.py')
| -rw-r--r-- | test_polymatrix/test_tomatrixrepr.py | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/test_polymatrix/test_tomatrixrepr.py b/test_polymatrix/test_tomatrixrepr.py index 88a35f2..dc023ca 100644 --- a/test_polymatrix/test_tomatrixrepr.py +++ b/test_polymatrix/test_tomatrixrepr.py @@ -1,9 +1,7 @@ import unittest -import polymatrix +import polymatrix.expression.initexpressionbase from polymatrix.expressionstate.init.initexpressionstate import init_expression_state -from polymatrix.expression.init.initfromtermsexpr import init_from_terms_expr -from polymatrix.expression.init.initlinearinexpr import init_linear_in_expr class TestToMatrixRepr(unittest.TestCase): @@ -29,12 +27,12 @@ class TestToMatrixRepr(unittest.TestCase): (1, 0): {((1, 1),): 1}, } - expr = init_from_terms_expr(terms=underlying_terms, shape=(3, 1)) + expr = polymatrix.expression.initexpressionbase.init_from_terms_expr(terms=underlying_terms, shape=(3, 1)) state = init_expression_state(n_param=2) state, result = polymatrix.to_matrix_repr( (expr,), - init_from_terms_expr(terms=variable_terms, shape=(2, 1),), + polymatrix.expression.initexpressionbase.init_from_terms_expr(terms=variable_terms, shape=(2, 1),), ).apply(state) A0 = result.data[0][0] |