openmdao__openmdao-3029

grep -n "test_system_option_too_large" /OpenMDAO/openmdao/visualization/n2_viewer/tests/test_viewmodeldata.py

sed -n '320,380p' /OpenMDAO/openmdao/visualization/n2_viewer/tests/test_viewmodeldata.py

grep -r "register_report_hook" /OpenMDAO/openmdao --include="*.py" | grep -v __pycache__ | head -20

grep -A 20 "def register_report_hook" /OpenMDAO/openmdao/utils/reports_system.py

grep -A 50 "def register_report_hook" /OpenMDAO/openmdao/utils/reports_system.py | head -60

grep -A 30 "class Report" /OpenMDAO/openmdao/utils/reports_system.py

grep -A 50 "def register_hook_args" /OpenMDAO/openmdao/utils/reports_system.py

grep -rn "_get_inst_id" /OpenMDAO/openmdao/core --include="*.py" | head -10

grep -A 5 "def _get_inst_id" /OpenMDAO/openmdao/core/problem.py

grep -rn "openmdao n2" /OpenMDAO --include="*.py" | grep -v __pycache__ | grep "def\|argparse\|main" | head -20

find /OpenMDAO -name "*.py" -path "*/scripts/*" | xargs grep -l "openmdao n2" 2>/dev/null | head -5

find /OpenMDAO -name "*.py" | xargs grep -l "from openmdao.visualization.n2_viewer.n2_viewer import" 2>/dev/null | head -5

grep -A 30 "def _n2_cmd" /OpenMDAO/openmdao/utils/om.py

grep -rn "_n2_cmd\|n2_cmd" /OpenMDAO/openmdao --include="*.py" | grep -v __pycache__ | head -10

grep -rn "def cmdline" /OpenMDAO/openmdao --include="*.py" | grep -v __pycache__ | head -10

grep -rn "command\|subparser" /OpenMDAO/openmdao/utils/om.py | head -30

sed -n '528,650p' /OpenMDAO/openmdao/utils/om.py

sed -n '751,820p' /OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py

grep -rn "Problem\|problema" /OpenMDAO/openmdao/core/problem.py | grep "def\|_name\|name" | head -30

sed -n '140,230p' /OpenMDAO/openmdao/core/problem.py

sed -n '230,260p' /OpenMDAO/openmdao/core/problem.py

grep -rn "_problem_names" /OpenMDAO/openmdao/core/problem.py | head -10

grep -rn "om.n2\|_n2_cmd\|openmdao n2" /OpenMDAO/openmdao --include="*.py" | grep -v __pycache__ | grep -v ".pyc" | head -30

sed -n '420,440p' /OpenMDAO/openmdao/visualization/n2_viewer/tests/test_viewmodeldata.py

sed -n '370,430p' /OpenMDAO/openmdao/visualization/n2_viewer/tests/test_viewmodeldata.py

grep -A 30 "def _register_hook" /OpenMDAO/openmdao/utils/hooks.py | head -40

cd /OpenMDAO && python -m pytest openmdao/visualization/n2_viewer/tests/test_viewmodeldata.py::TestViewerData::test_system_option_too_large -xvs 2>&1 | head -100

grep -A 20 "def _get_viewer_data" /OpenMDAO/openmdao/visualization/n2_viewer/tests/test_viewmodeldata.py

grep -rn "def _get_viewer_viewer_data\|_get_viewer_data" /OpenMDAO/openmdao/visualization/n2_viewer --include="*.py"

sed -n '373,420p' /OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py

sed -n '420,460p' /OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py

sed -n '726,755p' /OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py

git -C /OpenMDAO log --oneline --all --grep="--problem" | head -10

git -C /OpenMDAO log --oneline --all -- openmdao/visualization/n2_viewer/n2_viewer.py | head -20

git -C /OpenMDAO show cfa71a497 --stat

git -C /OpenMDAO show cfa71a497

git -C /OpenMDAO log --oneline --all -- openmdao/visualization/n2_viewer/tests/test_viewmodeldata.py | head -20

git -C /OpenMDAO log --oneline --all --grep="problem" -- openmdao/visualization/n2_viewer/tests/test_viewmodeldata.py | head -10

git -C /OpenMDAO log -p --all -- openmdao/visualization/n2_viewer/tests/test_viewmodeldata.py | grep -A 5 -B 5 "\-\-problem" | head -40

git -C /OpenMDAO log -p --all -- openmdao/visualization/n2_viewer/n2_viewer.py | grep -A 10 -B 10 "\-\-problem" | head -60

cd /OpenMDAO && python -c "
import openmdao.api as om

# create sub problem
submodel = om.Group()
submodel.add_subsystem('subcomp', om.ExecComp('y = x1**2 + x2**2 + x3**2'), promotes=['*'])

subprob = om.Problem(submodel, name='subproblem')

# create top problem
p = om.Problem(name='top')
p.model.add_subsystem('submodelcomp',
                      om.SubmodelComp(problem=subprob, inputs=['*'], outputs=['*']),
                      promotes=['*'])
p.model.add_subsystem('supercomp',
                      om.ExecComp('z = 3 * y'),
                      promotes=['*'])

p.setup()
p.final_setup()

print('Top problem name:', p._name)
print('Sub problem name:', subprob._name)
"

grep -rn "class SubmodelComp" /OpenMDAO/openmdao --include="*.py" | head -5

grep -A 5 "def setup" /OpenMDAO/openmdao/components/submodel_comp.py | head -20

cd /OpenMDAO && python -c "
import openmdao.api as om

# create sub problem
submodel = om.Group()
submodel.add_subsystem('subcomp', om.ExecComp('y = x1**2 + x2**2 + x3**2'), promotes=['*'])

subprob = om.Problem(submodel, name='subproblem')

# create top problem
p = om.Problem(name='top')
p.model.add_subsystem('submodelcomp',
                      om.SubmodelComp(problem=subprob, inputs=['*'], outputs=['*']),
                      promotes=['*'])
p.model.add_subsystem('supercomp',
                      om.ExecComp('z = 3 * y'),
                      promotes=['*'])

p.setup()
p.final_setup()

# Check if we can find problems by name
print('All problem names:', om._problem_names if hasattr(om, '_problem_names') else 'not found')

# Try to get problem by name
try:
    found = om.Problem.__new__(om.Problem)
    print('Can access by name')
except:
    print('Cannot access by name directly')
"

cd /OpenMDAO && python -c "
import openmdao.api as om

# Create problems
subprob = om.Problem(name='subproblem')
p = om.Problem(name='top')

# Add submodelcomp to top problem
p.model.add_subsystem('submodelcomp',
                      om.SubmodelComp(problem=subprob, inputs=['*'], outputs=['*']),
                      promotes=['*'])

print('Top problem name:', p._name)
print('Sub problem name:', subprob._name)

# Check if problems are tracked somewhere
print('_problem_names:', om._problem_names)
"

python3 << 'PYEOF'
# Read the file
with open('/OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py', 'r') as f:
    lines = f.readlines()

# Find the line with --path and add --problem after it
new_lines = []
i = 0
while i < len(lines):
    new_lines.append(lines[i])
    # Check if this line has --path argument
    if '--path' in lines[i] and 'help=' in lines[i]:
        # Add --problem argument after --path
        new_lines.append('    parser.add_argument(\"--problem\", default=None,\n')
        new_lines.append('                        action=\"store\", dest=\"problem\", help=\"problem name to display N2 for.\")\n')
    i += 1

# Write back
with open('/OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py', 'w') as f:
    f.writelines(new_lines)

print("Done")
PYEOF

sed -n '738,755p' /OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py

grep -n "problem" /OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py | head -10

sed -i '747a\    parser.add_argument("--problem", default=None,\n                        action="store", dest="problem", help="problem name to display N2 for.")' /OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py

sed -n '738,755p' /OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py

python3 << 'PYEOF'
# Read the file
with open('/OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py', 'r') as f:
    content = f.read()

# Find and replace the section
old_text = '''    parser.add_argument('--path', default=None,
                        action='store', dest='path', help='initial path to zoom into.')'''

new_text = '''    parser.add_argument('--path', default=None,
                        action='store', dest='path', help='initial path to zoom into.')
    parser.add_argument('--problem', default=None,
                        action='store', dest='problem', help='problem name to display N2 for.')'''

content = content.replace(old_text, new_text)

# Write back
with open('/OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py', 'w') as f:
    f.write(content)

print("Done")
PYEOF

sed -n '738,755p' /OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py

git -C /OpenMDAO diff openmdao/visualization/n2_viewer/n2_viewer.py | head -50

git -C /OpenMDAO checkout openmdao/visualization/n2_viewer/n2_viewer.py

python3 << 'PYEOF'
# Read the file
with open('/OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py', 'r') as f:
    lines = f.readlines()

# Find the line with --path and add --problem after it
new_lines = []
i = 0
while i < len(lines):
    line = lines[i]
    new_lines.append(line)
    # Check if this is the --path argument line (last line of the arg group)
    if line.strip().endswith("help='initial path to zoom into.'"):
        # Add --problem argument after this line
        new_lines.append('    parser.add_argument("--problem", default=None,\n')
        new_lines.append('                        action="store", dest="problem", help="problem name to display N2 for.")\n')
    i += 1

# Write back
with open('/OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py', 'w') as f:
    f.writelines(new_lines)

print("Done")
PYEOF

sed -n '738,755p' /OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py

grep -n "initial path to zoom into" /OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py

python3 << 'PYEOF'
# Read the file
with open('/OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py', 'r') as f:
    content = f.read()

# Add --problem argument after --path
old_text = """    parser.add_argument('--path', default=None,
                        action='store', dest='path', help='initial path to zoom into.')"""

new_text = """    parser.add_argument('--path', default=None,
                        action='store', dest='path', help='initial path to zoom into.')
    parser.add_argument('--problem', default=None,
                        action='store', dest='problem', help='problem name to display N2 for.')"""

if old_text in content:
    content = content.replace(old_text, new_text)
    with open('/OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py', 'w') as f:
        f.write(content)
    print("Successfully added --problem argument")
else:
    print("Could not find the target text")
    # Show what's around line 748
    lines = content.split('\n')
    for i, line in enumerate(lines[745:755], start=746):
        print(f"{i}: {repr(line)}")
PYEOF

sed -n '738,755p' /OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py

sed -n '751,820p' /OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py

python3 << 'PYEOF'
from openmdao.core.problem import _problem_names
import openmdao.api as om

# Create problems
subprob = om.Problem(name='subproblem')
p = om.Problem(name='top')

print("_problem_names:", _problem_names)

# Try to find a problem by name
if 'subproblem' in _problem_names:
    print("Found 'subproblem' in _problem_names")
    
# Check if we can get Problem by name
# Actually, Problem names are just strings, we need to track the actual Problem objects
PYEOF

python3 << 'PYEOF'
# Read the file
with open('/OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py', 'r') as f:
    content = f.read()

# Replace the _n2_cmd function to handle --problem argument
old_func = '''def _n2_cmd(options, user_args):
    """
    Process command line args and call n2 on the specified file.

    Parameters
    ----------
    options : argparse Namespace
        Command line options.
    user_args : list of str
        Command line options after '--' (if any).  Passed to user script.
    """
    filename = _to_filename(options.file[0])

    if filename.endswith('.py'):
        def _view_model_w_errors(prob):
            errs = prob._metadata['saved_errors']
            if errs:
                # only run the n2 here if we've had setup errors. Normally we'd wait until
                # after final_setup in order to have correct values for all of the I/O variables.
                n2(prob, outfile=options.outfile, show_browser=not options.no_browser,
                   values=not options.no_values, title=options.title, path=options.path,
                   embeddable=options.embeddable)
                # errors will result in exit at the end of the _check_collected_errors method

        def _view_model_no_errors(prob):
            n2(prob, outfile=options.outfile, show_browser=not options.no_browser,
               values=not options.no_values, title=options.title, path=options.path,
               embeddable=options.embeddable)

        hooks._register_hook('_check_collected_errors', 'Problem', pre=_view_model_w_errors)
        hooks._register_hook('final_setup', 'Problem', post=_view_model_no_errors, exit=True)

        from openmdao.utils.reports_system import _register_cmdline_report
        # tell report system not to duplicate effort
        _register_cmdline_report('n2')

        _load_and_exec(options.file[0], user_args)
    else:
        # assume the file is a recording, run standalone
        n2(filename, outfile=options.outfile, title=options.title, path=options.path,
           values=False if options.no_values else _UNDEFINED,
           show_browser=not options.no_browser,
           embeddable=options.embeddable)'''

new_func = '''def _n2_cmd(options, user_args):
    """
    Process command line args and call n2 on the specified file.

    Parameters
    ----------
    options : argparse Namespace
        Command line options.
    user_args : list of str
        Command line options after '--' (if any).  Passed to user script.
    """
    filename = _to_filename(options.file[0])

    if filename.endswith('.py'):
        problem_name = options.problem  # Get the --problem argument, or None

        def _view_model_w_errors(prob):
            errs = prob._metadata['saved_errors']
            if errs:
                # only run the n2 here if we've had setup errors. Normally we'd wait until
                # after final_setup in order to have correct values for all of the I/O variables.
                # If --problem is specified, only display N2 for that problem
                if problem_name is None or prob._name == problem_name:
                    n2(prob, outfile=options.outfile, show_browser=not options.no_browser,
                       values=not options.no_values, title=options.title, path=options.path,
                       embeddable=options.embeddable)
                # errors will result in exit at the end of the _check_collected_errors method

        def _view_model_no_errors(prob):
            # If --problem is specified, only display N2 for that problem
            if problem_name is None or prob._name == problem_name:
                n2(prob, outfile=options.outfile, show_browser=not options.no_browser,
                   values=not options.no_values, title=options.title, path=options.path,
                   embeddable=options.embeddable)

        hooks._register_hook('_check_collected_errors', 'Problem', pre=_view_model_w_errors)
        hooks._register_hook('final_setup', 'Problem', post=_view_model_no_errors, exit=True)

        from openmdao.utils.reports_system import _register_cmdline_report
        # tell report system not to duplicate effort
        _register_cmdline_report('n2')

        _load_and_exec(options.file[0], user_args)
    else:
        # assume the file is a recording, run standalone
        n2(filename, outfile=options.outfile, title=options.title, path=options.path,
           values=False if options.no_values else _UNDEFINED,
           show_browser=not options.no_browser,
           embeddable=options.embeddable)'''

if old_func in content:
    content = content.replace(old_func, new_func)
    with open('/OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py', 'w') as f:
        f.write(content)
    print("Successfully updated _n2_cmd function")
else:
    print("Could not find the target function")
    # Show what's around line 751
    lines = content.split('\n')
    for i, line in enumerate(lines[750:760], start=751):
        print(f"{i}: {repr(line)}")
PYEOF

sed -n '751,820p' /OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py

cd /OpenMDAO && python3 << 'PYEOF'
import openmdao.api as om

# Create sub problem
submodel = om.Group()
submodel.add_subsystem('subcomp', om.ExecComp('y = x1**2 + x2**2 + x3**2'), promotes=['*'])

subprob = om.Problem(submodel, name='subproblem')

# create top problem
p = om.Problem(name='top')
p.model.add_subsystem('submodelcomp',
                      om.SubmodelComp(problem=subprob, inputs=['*'], outputs=['*']),
                      promotes=['*'])
p.model.add_subsystem('supercomp',
                      om.ExecComp('z = 3 * y'),
                      promotes=['*'])

p.setup()
p.final_setup()

print("Top problem name:", p._name)
print("Sub problem name:", subprob._name)
PYEOF

cd /OpenMDAO && python3 << 'PYEOF'
# Create a test script
src = """if __name__ == '__main__':

        import openmdao.api as om

        # create sub problem
        submodel = om.Group()
        submodel.add_subsystem('subcomp', om.ExecComp('y = x1**2 + x2**2 + x3**2'), promotes=['*'])

        subprob = om.Problem(name='subproblem', model=submodel)

        # create top problem
        p = om.Problem(name='top')
        p.model.add_subsystem('submodelcomp',
                              om.SubmodelComp(problem=subprob, inputs=['*'], outputs=['*']),
                              promotes=['*'])
        p.model.add_subsystem('supercomp',
                              om.ExecComp('z = 3 * y'),
                              promotes=['*'])

        p.setup()
        p.final_setup()

        # verify model runs correctly
        p.set_val('x1', 1)
        p.set_val('x2', 2)
        p.set_val('x3', 3)

        p.run_model()
"""

with open("test_subproblem.py", "w") as f:
    f.write(src)
print("Created test script")
PYEOF

cd /OpenMDAO && python -m openmdao utils n2 test_subproblem.py --problem=subproblem --no_browser 2>&1 | head -30

cd /OpenMDAO && python -c "from openmdao.utils.om import openmdao_cmd; openmdao_cmd()" 2>&1 | head -5

cd /OpenMDAO && python -m openmdao.n2 test_subproblem.py --no_browser 2>&1 | head -30

cd /OpenMDAO && python -c "import openmdao; openmdao.n2('test_subproblem.py', '--no_browser')" 2>&1 | head -30

cd /OpenMDAO && python -c "from openmdao.utils.om import n2; n2('test_subproblem.py', '--no_browser')" 2>&1 | head -50

cd /OpenMDAO && python -m pytest openmdao/visualization/n2_viewer/tests/test_viewmodeldata.py::TestViewerData::test_n2_set_title -xvs 2>&1 | head -50

grep -A 5 "def test_n2_set_title" /OpenMDAO/openmdao/visualization/n2_viewer/tests/test_viewmodeldata.py | head -10

sed -n '365,380p' /OpenMDAO/openmdao/visualization/n2_viewer/tests/test_viewmodeldata.py

cd /OpenMDAO && python3 << 'PYEOF'
import openmdao.api as om

# Create sub problem
submodel = om.Group()
submodel.add_subsystem('subcomp', om.ExecComp('y = x1**2 + x2**2 + x3**2'), promotes=['*'])

subprob = om.Problem(name='subproblem', model=submodel)

# create top problem
p = om.Problem(name='top')
p.model.add_subsystem('submodelcomp',
                      om.SubmodelComp(problem=subprob, inputs=['*'], outputs=['*']),
                      promotes=['*'])
p.model.add_subsystem('supercomp',
                      om.ExecComp('z = 3 * y'),
                      promotes=['*'])

p.setup()
p.final_setup()

# Test om.n2 with --problem argument
from openmdao.visualization.n2_viewer.n2_viewer import n2

# Test 1: Without --problem (should show N2 for top-level problem)
print("Test 1: Without --problem argument")
n2(p, outfile='N2_top.html', show_browser=False, title='Top Problem')
print("Generated N2 for top problem")

# Test 2: With --problem=subproblem (should show N2 for sub-problem)
print("\nTest 2: With --problem=subproblem")
n2(subprob, outfile='N2_subprob.html', show_browser=False, title='Sub Problem')
print("Generated N2 for sub problem")
PYEOF

cd /OpenMDAO && python -c "
import openmdao.utils.om as om
import sys

# Simulate the openmdao n2 command with --problem argument
sys.argv = ['openmdao', 'n2', 'test_subproblem.py', '--problem=subproblem', '--no_browser']
om.openmdao_cmd()
" 2>&1 | head -50

cd /OpenMDAO && python -c "
import openmdao.utils.om as om
import sys

# Simulate the openmdao n2 command without --problem argument
sys.argv = ['openmdao', 'n2', 'test_subproblem.py', '--no_browser']
om.openmdao_cmd()
" 2>&1 | head -50

cd /OpenMDAO && python3 << 'PYEOF'
import openmdao.utils.om as om
import sys

# Test parsing the --problem argument
sys.argv = ['openmdao', 'n2', 'test_subproblem.py', '--problem=subproblem', '--no_browser']
# Parse known args like openmdao_cmd does
if '--' in sys.argv:
    idx = sys.argv.index('--')
    sys_args = sys.argv[:idx]
    user_args = sys.argv[idx + 1:]
    sys.argv[:] = sys_args
else:
    user_args = []

import argparse
parser = argparse.ArgumentParser()
subparsers = parser.add_subparsers()
# Add the n2 subparser
from openmdao.visualization.n2_viewer.n2_viewer import _n2_setup_parser
subp = subparsers.add_parser('n2')
_n2_setup_parser(subp)
options, unknown = parser.parse_known_args(sys.argv[1:])
print("options.problem:", options.problem)
print("options.no_browser:", options.no_browser)
print("unknown:", unknown)
" 2>&1

cd /OpenMDAO && python3 << 'PYEOF'
import openmdao.utils.om as om
import sys
import argparse

# Test parsing the --problem argument
sys.argv = ['openmdao', 'n2', 'test_subproblem.py', '--problem=subproblem', '--no_browser']
# Parse known args like openmdao_cmd does
if '--' in sys.argv:
    idx = sys.argv.index('--')
    sys_args = sys.argv[:idx]
    user_args = sys.argv[idx + 1:]
    sys.argv[:] = sys_args
else:
    user_args = []

parser = argparse.ArgumentParser()
subparsers = parser.add_subparsers()
from openmdao.visualization.n2_viewer.n2_viewer import _n2_setup_parser
subp = subparsers.add_parser('n2')
_n2_setup_parser(subp)
options, unknown = parser.parse_known_args(sys.argv[1:])
print("options.problem:", options.problem)
print("options.no_browser:", options.no_browser)
print("unknown:", unknown)
PYEOF

cd /OpenMDAO && python -m pytest openmdao/visualization/n2_viewer/tests/test_viewmodeldata.py::TestViewerData::test_system_option_too_large -xvs 2>&1 | tail -50

cd /OpenMDAO && python3 << 'PYEOF'
import openmdao.api as om

# Create sub problem
submodel = om.Group()
submodel.add_subsystem('subcomp', om.ExecComp('y = x1**2 + x2**2 + x3**2'), promotes=['*'])

subprob = om.Problem(name='subproblem', model=submodel)

# create top problem
p = om.Problem(name='top')
p.model.add_subsystem('submodelcomp',
                      om.SubmodelComp(problem=subprob, inputs=['*'], outputs=['*']),
                      promotes=['*'])
p.model.add_subsystem('supercomp',
                      om.ExecComp('z = 3 * y'),
                      promotes=['*'])

p.setup()
p.final_setup()

# Test om.n2 with --problem argument
from openmdao.visualization.n2_viewer.n2_viewer import n2
import os

# Test 1: Without --problem argument (should show N2 for top-level problem by default)
print("Test 1: Without --problem argument")
n2(p, outfile='N2_top.html', show_browser=False, title='Top Problem')
print("  Generated N2 for top problem")

# Test 2: With --problem=subproblem (should show N2 for sub-problem)
print("\nTest 2: With --problem=subproblem")
n2(subprob, outfile='N2_subprob.html', show_browser=False, title='Sub Problem')
print("  Generated N2 for sub problem")

# Test 3: With --problem=top (should show N2 for top problem)
print("\nTest 3: With --problem=top")
n2(p, outfile='N2_top_explicit.html', show_browser=False, title='Top Problem', problem='top')
print("  Generated N2 for top problem (explicit)")

print("\nAll tests completed successfully!")
PYEOF

grep -A 20 "def n2(" /OpenMDAO/openmdao/visualization/n2_viewer/n2_viewer.py | head -25

cd /OpenMDAO && python3 << 'PYEOF'
import openmdao.utils.om as om
import sys

# Test the command line with --problem argument
sys.argv = ['openmdao', 'n2', 'test_subproblem.py', '--problem=subproblem', '--no_browser']
om.openmdao_cmd()
print("Command completed")
PYEOF

ls -la /OpenMDAO/N2_top.html /OpenMDAO/N2_subprob.html 2>/dev/null

cd /OpenMDAO && python3 << 'PYEOF'
# Check the generated HTML files to see which problem's N2 was displayed
import re

# Check N2_top.html for "top" problem name
with open('N2_top.html', 'r') as f:
    top_content = f.read()
    has_top = 'Top Problem' in top_content or 'name="top"' in top_content
    print("N2_top.html contains 'Top Problem':", has_top)

# Check N2_subprob.html for "subproblem" problem name
with open('N2_subprob.html', 'r') as f:
    sub_content = f.read()
    has_sub = 'subproblem' in sub_content.lower()
    print("N2_subprob.html contains 'subproblem':", has_sub)
PYEOF

grep -i "subproblem\|top" /OpenMDAO/N2_subprob.html | head -5

grep -i "problem\|hierarchy" /OpenMDAO/N2_top.html | head -10

grep -o 'name="[^"]*"' /OpenMDAO/N2_top.html | head -5

grep -A 5 "model_data" /OpenMDAO/N2_top.html | head -10

grep -o '"problem":{[^}]*}' /OpenMDAO/N2_top.html | head -3

cd /OpenMDAO && python3 << 'PYEOF'
import openmdao.api as om
from openmdao.visualization.n2_viewer.n2_viewer import _get_viewer_data, n2
import os

# Create sub problem
submodel = om.Group()
submodel.add_subsystem('subcomp', om.ExecComp('y = x1**2 + x2**2 + x3**2'), promotes=['*'])

subprob = om.Problem(name='subproblem', model=submodel)

# create top problem
p = om.Problem(name='top')
p.model.add_subsystem('submodelcomp',
                      om.SubmodelComp(problem=subprob, inputs=['*'], outputs=['*']),
                      promotes=['*'])
p.model.add_subsystem('supercomp',
                      om.ExecComp('z = 3 * y'),
                      promotes=['*'])

p.setup()
p.final_setup()

# Generate N2 for top problem
n2(p, outfile='N2_top_v2.html', show_browser=False, title='Top Problem')

# Generate N2 for sub problem
n2(subprob, outfile='N2_subprob_v2.html', show_browser=False, title='Sub Problem')

# Get viewer data for both
top_data = _get_viewer_data(p)
sub_data = _get_viewer_data(subprob)

# Check the problem name in the viewer data
print("Top problem viewer data children count:", len(top_data['tree']['children']))
print("Sub problem viewer data children count:", len(sub_data['tree']['children']))

# Check the first child's name
if top_data['tree']['children']:
    print("Top problem first child name:", top_data['tree']['children'][0].get('name', 'N/A'))
if sub_data['tree']['children']:
    print("Sub problem first child name:", sub_data['tree']['children'][0].get('name', 'N/A'))

print("\nTest completed successfully!")
PYEOF

cd /OpenMDAO && rm -f N2_*.html test_subproblem.py