Skip to content

Commit

Permalink
black reformat new unittest
Browse files Browse the repository at this point in the history
  • Loading branch information
sean-engelstad committed Nov 2, 2023
1 parent 4b4b306 commit 92275e9
Showing 1 changed file with 8 additions and 6 deletions.
14 changes: 8 additions & 6 deletions tests/unit_tests/framework/test_functions_file.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,8 +34,8 @@

remote = Remote(comm, ".py", base_dir, nprocs=1)

class TestFunctionsFile(unittest.TestCase):

class TestFunctionsFile(unittest.TestCase):
def test_multiscenario_aeroelastic(self):
"""
test composite functions over two scenarios with aerothermoelastic coupling and a funtofem driver
Expand Down Expand Up @@ -93,9 +93,7 @@ def test_multiscenario_aeroelastic(self):
solvers.flow = TestAerodynamicSolver(comm, model)

# instantiate the driver to in initialize transfer scheme
FUNtoFEMnlbgs(
solvers, transfer_settings=TransferSettings(npts=5), model=model
)
FUNtoFEMnlbgs(solvers, transfer_settings=TransferSettings(npts=5), model=model)

# set analysis functions to random values
for func in model.get_functions():
Expand All @@ -120,10 +118,14 @@ def test_multiscenario_aeroelastic(self):
model.evaluate_composite_functions(compute_grad=True)

# write out the functions file again
model.write_functions_file(comm, remote._functions_file+"2", full_precision=True)
model.write_functions_file(
comm, remote._functions_file + "2", full_precision=True
)

# write the full functions file for optimization
model.write_functions_file(comm, remote.functions_file, full_precision=False, optim=True)
model.write_functions_file(
comm, remote.functions_file, full_precision=False, optim=True
)


if __name__ == "__main__":
Expand Down

0 comments on commit 92275e9

Please sign in to comment.