From 9e88e706fe92860ce973630a4a288b15a4a8b66c Mon Sep 17 00:00:00 2001
From: Jonathan Shimwell <mail@jshimwell.com>
Date: Thu, 16 Feb 2023 16:20:44 +0000
Subject: [PATCH] testing get_all_universes is in DAGMCUniverse

---
 tests/unit_tests/test_universe.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/tests/unit_tests/test_universe.py b/tests/unit_tests/test_universe.py
index 26b7779a70..a70bd46673 100644
--- a/tests/unit_tests/test_universe.py
+++ b/tests/unit_tests/test_universe.py
@@ -93,10 +93,12 @@ def test_get_all_universes():
     u2 = openmc.Universe(cells=[c2])
     c3 = openmc.Cell(fill=u1)
     c4 = openmc.Cell(fill=u2)
-    u3 = openmc.Universe(cells=[c3, c4])
+    u3 = openmc.DAGMCUniverse(filename="")
+    c5 = openmc.Cell(fill=u3)
+    u4 = openmc.Universe(cells=[c3, c4, c5])
 
-    univs = set(u3.get_all_universes().values())
-    assert not (univs ^ {u1, u2})
+    univs = set(u4.get_all_universes().values())
+    assert not (univs ^ {u1, u2, u3})
 
 
 def test_clone():