diff --git a/dymos/examples/brachistochrone/test/test_doc_brachistochrone_polynomial_controls.py b/dymos/examples/brachistochrone/test/test_doc_brachistochrone_polynomial_controls.py index 5c3f2f48c..11fd4881e 100644 --- a/dymos/examples/brachistochrone/test/test_doc_brachistochrone_polynomial_controls.py +++ b/dymos/examples/brachistochrone/test/test_doc_brachistochrone_polynomial_controls.py @@ -611,7 +611,7 @@ def test_brachistochrone_polynomial_control_gauss_lobatto(self): for dymos_type, prom_names in map_type_to_promnames.items(): prom_outputs = {meta['prom_name'] for abs_path, meta in - ts_group.list_outputs(tags=[dymos_type], out_stream=None)} + ts_group.list_outputs(tags=[dymos_type], prom_name=True, out_stream=None)} self.assertSetEqual(prom_outputs, prom_names, msg=f'\n{dymos_type}\nin outputs: {prom_outputs}\nexpected: {prom_names}') @@ -691,7 +691,7 @@ def test_brachistochrone_polynomial_control_radau(self): for dymos_type, prom_names in map_type_to_promnames.items(): prom_outputs = {meta['prom_name'] for abs_path, meta in - ts_group.list_outputs(tags=[dymos_type], out_stream=None)} + ts_group.list_outputs(tags=[dymos_type], prom_name=True, out_stream=None)} self.assertSetEqual(prom_outputs, prom_names, msg=f'\n{dymos_type}\nin outputs: {prom_outputs}\nexpected: {prom_names}')