1 files changed, 19 insertions, 18 deletions
diff --git a/test/dl/test_c_union_entity.rb b/test/dl/test_c_union_entity.rb
index 29b9e1054a..09f7c60e4c 100644
--- a/test/dl/test_c_union_entity.rb
+++ b/test/dl/test_c_union_entity.rb
@@ -2,29 +2,30 @@ require_relative 'test_base'
 
 require 'dl/struct'
 
-class DL::TestCUnionEntity < DL::TestBase
-  def test_class_size
-    size = DL::CUnionEntity.size([DL::TYPE_DOUBLE, DL::TYPE_CHAR])
+module DL
+  class TestCUnionEntity < TestBase
+    def test_class_size
+      size = CUnionEntity.size([TYPE_DOUBLE, TYPE_CHAR])
 
-    assert_equal DL::SIZEOF_DOUBLE, size
-  end
+      assert_equal SIZEOF_DOUBLE, size
+    end
 
-  def test_class_size_with_count
-    size = DL::CUnionEntity.size([[DL::TYPE_DOUBLE, 2], [DL::TYPE_CHAR, 20]])
+    def test_class_size_with_count
+      size = CUnionEntity.size([[TYPE_DOUBLE, 2], [TYPE_CHAR, 20]])
 
-    assert_equal DL::SIZEOF_CHAR * 20, size
-  end
+      assert_equal SIZEOF_CHAR * 20, size
+    end
 
-  def test_set_ctypes
-    union = DL::CUnionEntity.malloc [DL::TYPE_INT, DL::TYPE_LONG]
-    union.assign_names %w[int long]
+    def test_set_ctypes
+      union = CUnionEntity.malloc [TYPE_INT, TYPE_LONG]
+      union.assign_names %w[int long]
 
-    # this test is roundabout because the stored ctypes are not accessible
-    union['long'] = 1
-    assert_equal 1, union['long']
+      # this test is roundabout because the stored ctypes are not accessible
+      union['long'] = 1
+      assert_equal 1, union['long']
 
-    union['int'] = 1
-    assert_equal 1, union['int']
+      union['int'] = 1
+      assert_equal 1, union['int']
+    end
   end
 end
-