summaryrefslogtreecommitdiffstats
path: root/test_polymatrix/test_expression/test_truncate.py
diff options
context:
space:
mode:
Diffstat (limited to 'test_polymatrix/test_expression/test_truncate.py')
-rw-r--r--test_polymatrix/test_expression/test_truncate.py11
1 files changed, 4 insertions, 7 deletions
diff --git a/test_polymatrix/test_expression/test_truncate.py b/test_polymatrix/test_expression/test_truncate.py
index d6cafbe..c73e98d 100644
--- a/test_polymatrix/test_expression/test_truncate.py
+++ b/test_polymatrix/test_expression/test_truncate.py
@@ -1,10 +1,7 @@
import unittest
from polymatrix.expressionstate.init.initexpressionstate import init_expression_state
-from polymatrix.expression.init.initfromtermsexpr import init_from_terms_expr
-from polymatrix.expression.init.initquadraticinexpr import init_quadratic_in_expr
-from polymatrix.expression.init.initsymmetricexpr import init_symmetric_expr
-from polymatrix.expression.init.inittruncateexpr import init_truncate_expr
+import polymatrix.expression.initexpressionbase
class TestTruncate(unittest.TestCase):
@@ -30,9 +27,9 @@ class TestTruncate(unittest.TestCase):
}
- expr = init_truncate_expr(
- underlying=init_from_terms_expr(terms=terms, shape=(2, 2)),
- variables=init_from_terms_expr(terms=variable_terms, shape=(2, 1),),
+ expr = polymatrix.expression.initexpressionbase.init_truncate_expr(
+ underlying=polymatrix.expression.initexpressionbase.init_from_terms_expr(terms=terms, shape=(2, 2)),
+ variables=polymatrix.expression.initexpressionbase.init_from_terms_expr(terms=variable_terms, shape=(2, 1),),
degrees=(1, 2),
)