Slight test refactoring (#181)

BayesWitnesses · Mar 18, 2020 · d058b8f · d058b8f
1 parent 46c6ab2
commit d058b8f
Showing 1 changed file with 8 additions and 12 deletions.
diff --git a/tests/assemblers/test_svm.py b/tests/assemblers/test_svm.py
@@ -108,28 +108,24 @@ def test_cosine_kernel():
     actual = assembler.assemble()
 
     def kernel_ast(sup_vec_value):
+        feature_norm = ast.SqrtExpr(
+            ast.BinNumExpr(
+                ast.FeatureRef(0),
+                ast.FeatureRef(0),
+                ast.BinNumOpType.MUL),
+            to_reuse=True)
         return ast.BinNumExpr(
             ast.BinNumExpr(
                 ast.NumVal(sup_vec_value),
                 ast.FeatureRef(0),
                 ast.BinNumOpType.MUL),
             ast.IfExpr(
                 ast.CompExpr(
-                    ast.SqrtExpr(
-                        ast.BinNumExpr(
-                            ast.FeatureRef(0),
-                            ast.FeatureRef(0),
-                            ast.BinNumOpType.MUL),
-                        to_reuse=True),
+                    feature_norm,
                     ast.NumVal(0.0),
                     ast.CompOpType.EQ),
                 ast.NumVal(1.0),
-                ast.SqrtExpr(
-                    ast.BinNumExpr(
-                        ast.FeatureRef(0),
-                        ast.FeatureRef(0),
-                        ast.BinNumOpType.MUL),
-                    to_reuse=True)),
+                feature_norm),
             ast.BinNumOpType.DIV)
 
     expected = _create_expected_single_output_ast(