import os
from subprocess import check_call

import pytest


def run(command, problem, so, shape, nbpml, *extra):
    args = ["python", "../benchmarks/user/benchmark.py", command]
    args.extend(["-P", str(problem)])
    args.extend(["-so", str(so)])
    args.extend(["-d"] + [str(i) for i in shape])
    args.extend(["--nbpml", str(nbpml)])
    args.extend(extra)
    check_call(args)


def test_test_tti():
    run('test', 'tti', 4, [20, 20, 20], 5)


def test_test_acoustic():
    run('test', 'acoustic', 4, [20, 20, 20], 5)


def test_run_acoustic_fixed_bs():
    run('run', 'acoustic', 4, [20, 20, 20], 5, '-bs', '5', '5', '4')

from subprocess import check_call


def run_cmd(command, problem, so, shape, nbpml, *extra):
    args = ["python", "../benchmarks/user/benchmark.py", command]
    args.extend(["-P", str(problem)])
    args.extend(["-so", str(so)])
    args.extend(["-d"] + [str(i) for i in shape])
    args.extend(["--nbpml", str(nbpml)])
    args.extend(extra)
    check_call(args)


def test_test_tti():
    run_cmd('test', 'tti', 4, [20, 20, 20], 5)


def test_test_acoustic():
    run_cmd('test', 'acoustic', 4, [20, 20, 20], 5)


def test_run_acoustic_fixed_bs():
    run_cmd('run', 'acoustic', 4, [20, 20, 20], 5, '-bs', '5', '5', '4')

Check bench mode==run with fixed block shape
tests: Check bench mode==run with fixed block shape

- import os
  from subprocess import check_call
  
- import pytest
  
- 
- def run(command, problem, so, shape, nbpml, *extra):
+ def run_cmd(command, problem, so, shape, nbpml, *extra):
      args = ["python", "../benchmarks/user/benchmark.py", command]
      args.extend(["-P", str(problem)])
      args.extend(["-so", str(so)])
      args.extend(["-d"] + [str(i) for i in shape])
      args.extend(["--nbpml", str(nbpml)])
      args.extend(extra)
      check_call(args)
  
  
  def test_test_tti():
-     run('test', 'tti', 4, [20, 20, 20], 5)
+     run_cmd('test', 'tti', 4, [20, 20, 20], 5)
  
  
  def test_test_acoustic():
-     run('test', 'acoustic', 4, [20, 20, 20], 5)
+     run_cmd('test', 'acoustic', 4, [20, 20, 20], 5)
  
  
  def test_run_acoustic_fixed_bs():
-     run('run', 'acoustic', 4, [20, 20, 20], 5, '-bs', '5', '5', '4')
+     run_cmd('run', 'acoustic', 4, [20, 20, 20], 5, '-bs', '5', '5', '4')
  
Check bench mode==run with fixed block shape
## Code Before:
import os
from subprocess import check_call

import pytest


def run(command, problem, so, shape, nbpml, *extra):
    args = ["python", "../benchmarks/user/benchmark.py", command]
    args.extend(["-P", str(problem)])
    args.extend(["-so", str(so)])
    args.extend(["-d"] + [str(i) for i in shape])
    args.extend(["--nbpml", str(nbpml)])
    args.extend(extra)
    check_call(args)


def test_test_tti():
    run('test', 'tti', 4, [20, 20, 20], 5)


def test_test_acoustic():
    run('test', 'acoustic', 4, [20, 20, 20], 5)


def test_run_acoustic_fixed_bs():
    run('run', 'acoustic', 4, [20, 20, 20], 5, '-bs', '5', '5', '4')

## Instruction:
Check bench mode==run with fixed block shape
## Code After:
from subprocess import check_call


def run_cmd(command, problem, so, shape, nbpml, *extra):
    args = ["python", "../benchmarks/user/benchmark.py", command]
    args.extend(["-P", str(problem)])
    args.extend(["-so", str(so)])
    args.extend(["-d"] + [str(i) for i in shape])
    args.extend(["--nbpml", str(nbpml)])
    args.extend(extra)
    check_call(args)


def test_test_tti():
    run_cmd('test', 'tti', 4, [20, 20, 20], 5)


def test_test_acoustic():
    run_cmd('test', 'acoustic', 4, [20, 20, 20], 5)


def test_run_acoustic_fixed_bs():
    run_cmd('run', 'acoustic', 4, [20, 20, 20], 5, '-bs', '5', '5', '4')
