Skip to content
Snippets Groups Projects

Suggested style fixes for data fixtures

Merged Andreas Klöckner requested to merge fix-mr-29 into data-fixtures
All threads resolved!
3 files
+ 37
16
Compare changes
  • Side-by-side
  • Inline
Files
3
+ 7
5
import pytest
import utilities as u
# {{{ FluxDataSingle
class FluxDataSingle:
@@ -54,24 +55,25 @@ class FluxDataSingle:
def swap_array(self, arr, d):
p = self.permutation(d)
arr[p] = arr[[1,2,3]]
arr[p] = arr[[1, 2, 3]]
return arr
def swap_array_rows(self, arr, d):
p = self.permutation(d)
arr[p,:] = arr[[1,2,3],:]
arr[p, :] = arr[[1, 2, 3], :]
return arr
def swap_array_cols(self, arr, d):
p = self.permutation(d)
arr[:,p] = arr[:,[1,2,3]]
arr[:, p] = arr[:, [1, 2, 3]]
return arr
def permutation(self, d):
return [(d-1+i)%3 + 1 for i in range(3)]
return [(d-1+i) % 3 + 1 for i in range(3)]
# }}}
single_data = {}
# {{{ Input data: Case (a)
@@ -703,7 +705,7 @@ single_data["Case d:z"] = FluxDataSingle(
"Case b:x", "Case b:y", "Case b:z",
"Case c:x", "Case c:y", "Case c:z",
"Case d:x", "Case d:y", "Case d:z"])
def data(request):
def flux_test_data_fixture(request):
return single_data[request.param]
Loading