Update

jump-dev · Sep 20, 2023 · 645f292 · 645f292
1 parent 2101304
commit 645f292
Showing 1 changed file with 12 additions and 49 deletions.
diff --git 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