fixes after review

wpbonelli · Oct 30, 2024 · aab7e88 · aab7e88
1 parent 691f4f9
commit aab7e88
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 11 deletions.
diff --git a/autotest/test_codegen.py b/autotest/test_codegen.py
@@ -25,16 +25,6 @@ def test_dfn_load(dfn_name):
         common, _ = Dfn._load(common_file)
         dfn = Dfn.load(dfn_file, name=name, common=common)
 
-    if name in [
-        ("sln", "ems"),
-        ("exg", "gwfprt"),
-        ("exg", "gwfgwe"),
-        ("exg", "gwfgwt"),
-    ]:
-        assert not any(dfn)
-    else:
-        assert any(dfn)
-
 
 @pytest.mark.parametrize("dfn_name", DFN_NAMES)
 def test_make_targets(dfn_name, function_tmpdir):

diff --git a/flopy/mf6/utils/codegen/context.py b/flopy/mf6/utils/codegen/context.py
@@ -151,7 +151,7 @@ def from_dfn(dfn: Dfn) -> List["Context.Name"]:
                         Context.Name(*dfn.name),  # nam pkg
                         Context.Name(dfn.name.l, None),  # model
                     ]
-            elif (dfn.name.l, dfn.name.r) in [
+            elif dfn.name in [
                 ("gwf", "mvr"),
                 ("gwf", "gnc"),
                 ("gwt", "mvt"),