From 645f292be84e81c49b4084a4d1a1ffb8fb7ec7ae Mon Sep 17 00:00:00 2001
From: odow <o.dowson@gmail.com>
Date: Thu, 21 Sep 2023 10:23:11 +1200
Subject: [PATCH] Update

---
 test/rewrite_generic.jl | 61 ++++++++---------------------------------
 1 file changed, 12 insertions(+), 49 deletions(-)

diff --git a/test/rewrite_generic.jl b/test/rewrite_generic.jl
index 7b85874..0b752a7 100644
--- a/test/rewrite_generic.jl
+++ b/test/rewrite_generic.jl
@@ -345,55 +345,18 @@ function test_rewrite_expression()
 end
 
 function test_rewrite_generic_sum_dims()
-    x = [1 2; 3 4]
-    @test ==(
-        MA.@rewrite(sum(x; dims = 1), move_factors_into_sums = false),
-        [4 6],
-    )
-    @test ==(
-        MA.@rewrite(sum(x; dims = 2), move_factors_into_sums = false),
-        [3; 7;;],
-    )
-    @test ==(
-        MA.@rewrite(sum(x; dims = 1, init = 0), move_factors_into_sums = false),
-        [4 6],
-    )
-    @test ==(
-        MA.@rewrite(sum(x; dims = 2, init = 0), move_factors_into_sums = false),
-        [3; 7;;],
-    )
-    @test ==(
-        MA.@rewrite(sum(x; init = 0, dims = 1), move_factors_into_sums = false),
-        [4 6],
-    )
-    @test ==(
-        MA.@rewrite(sum(x; init = 0, dims = 2), move_factors_into_sums = false),
-        [3; 7;;],
-    )
-    @test ==(
-        MA.@rewrite(sum(x, dims = 1), move_factors_into_sums = false),
-        [4 6],
-    )
-    @test ==(
-        MA.@rewrite(sum(x, dims = 2), move_factors_into_sums = false),
-        [3; 7;;],
-    )
-    @test ==(
-        MA.@rewrite(sum(x, dims = 1, init = 0), move_factors_into_sums = false),
-        [4 6],
-    )
-    @test ==(
-        MA.@rewrite(sum(x, dims = 2, init = 0), move_factors_into_sums = false),
-        [3; 7;;],
-    )
-    @test ==(
-        MA.@rewrite(sum(x, init = 0, dims = 1), move_factors_into_sums = false),
-        [4 6],
-    )
-    @test ==(
-        MA.@rewrite(sum(x, init = 0, dims = 2), move_factors_into_sums = false),
-        [3; 7;;],
-    )
+    @test_rewrite sum([1 2; 3 4]; dims = 1)
+    @test_rewrite sum([1 2; 3 4]; dims = 2)
+    @test_rewrite sum([1 2; 3 4]; dims = 1, init = 0)
+    @test_rewrite sum([1 2; 3 4]; dims = 2, init = 0)
+    @test_rewrite sum([1 2; 3 4]; init = 0, dims = 1)
+    @test_rewrite sum([1 2; 3 4]; init = 0, dims = 2)
+    @test_rewrite sum([1 2; 3 4], dims = 1)
+    @test_rewrite sum([1 2; 3 4], dims = 2)
+    @test_rewrite sum([1 2; 3 4], dims = 1, init = 0)
+    @test_rewrite sum([1 2; 3 4], dims = 2, init = 0)
+    @test_rewrite sum([1 2; 3 4], init = 0, dims = 1)
+    @test_rewrite sum([1 2; 3 4], init = 0, dims = 2)
     return
 end