test_zero_mean_unit_variance_fixed

FynnBe · FynnBe · commit 6f78383228c8 · 2025-03-28T15:56:51.000+01:00
diff --git a/tests/test_proc_ops.py b/tests/test_proc_ops.py
@@ -89,7 +89,7 @@ def test_zero_mean_unit_variance_fixed(tid: MemberId):
         mean=xr.DataArray([3, 4, 5], dims=("channel",)),
         std=xr.DataArray([2.44948974, 2.44948974, 2.44948974], dims=("channel",)),
     )
-    data = xr.DataArray(np.arange(9).reshape((1, 3, 3)), dims=("b", "channel", "x"))
+    data = xr.DataArray(np.arange(9).reshape((1, 3, 3)), dims=("batch", "channel", "x"))
     expected = xr.DataArray(
         np.array(
             [
@@ -100,7 +100,7 @@ def test_zero_mean_unit_variance_fixed(tid: MemberId):
                 ]
             ]
         ),
-        dims=("b", "channel", "x"),
+        dims=("batch", "channel", "x"),
     )
     sample = Sample(members={tid: Tensor.from_xarray(data)}, stat={}, id=None)
     op(sample)

Original file line number	Diff line number	Diff line change
`@@ -89,7 +89,7 @@ def test_zero_mean_unit_variance_fixed(tid: MemberId):`
`89`	`89`	`mean=xr.DataArray([3, 4, 5], dims=("channel",)),`
`90`	`90`	`std=xr.DataArray([2.44948974, 2.44948974, 2.44948974], dims=("channel",)),`
`91`	`91`	`)`
`92`		`- data = xr.DataArray(np.arange(9).reshape((1, 3, 3)), dims=("b", "channel", "x"))`
	`92`	`+ data = xr.DataArray(np.arange(9).reshape((1, 3, 3)), dims=("batch", "channel", "x"))`
`93`	`93`	`expected = xr.DataArray(`
`94`	`94`	`np.array(`
`95`	`95`	`[`
`@@ -100,7 +100,7 @@ def test_zero_mean_unit_variance_fixed(tid: MemberId):`
`100`	`100`	`]`
`101`	`101`	`]`
`102`	`102`	`),`
`103`		`- dims=("b", "channel", "x"),`
	`103`	`+ dims=("batch", "channel", "x"),`
`104`	`104`	`)`
`105`	`105`	`sample = Sample(members={tid: Tensor.from_xarray(data)}, stat={}, id=None)`
`106`	`106`	`op(sample)`