lutaml 0.10.9 → 0.10.11

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c86294799b84e5b722bd271cba3e8f8bb47b5ca2b25a8e8a1fdfac763cc7ac42
-  data.tar.gz: c32950a48ea9681d0091b16f5df20e99c64c107703677467709a28353630cc92
+  metadata.gz: 60815cadb4a74fcb9fcdc3b84a3f6c19b450e5a6f1670d918ffd4d2e66d4a3a5
+  data.tar.gz: 2e24af4ab233104a171c7cac58a7416e79e5248ce61831d45ba930e44d14685b
 SHA512:
-  metadata.gz: 46ac28ea247ea946d5fa18d93824698d7b10756abb75b28992e148d22ea0e9c97e00a48b54eaa6929b1240a7732cca7edc60914edb243cbccb311de589611422
-  data.tar.gz: 83e71c5fd9dd4237743ca89894edbc6ecbad02d67d47f115d24e20f085531322b6e94cdcbede356665c35c2e6fec4ae81facafc1de689fa934ad1f1db06685eb
+  metadata.gz: 4ebc0be5174ba5dafb1fbda8aa962936d9c8de3b3d909d66844a83f35cf0f353c8131af4d45f945cdd1e9bfdf4975a52b09bbe17a18cf09eae571cc5eb297a83
+  data.tar.gz: e41165b1694af4e86a4b2373643ec94d6bc6c0a84310c2573ba001ad5dbc278917ba796f9e2a95174fdd1b12607fb68b4fbd487f46032b2a92b8850ff301d3ae

data/.rubocop_todo.yml

@@ -1,12 +1,12 @@
 # This configuration was generated by
 # `rubocop --auto-gen-config`
-# on 2026-05-08 06:29:20 UTC using RuboCop version 1.86.1.
+# on 2026-05-11 09:50:12 UTC using RuboCop version 1.86.1.
 # The point is for the user to remove these configuration records
 # one by one as the offenses are removed from the code base.
 # Note that changes in the inspected code, or installation of new
 # versions of RuboCop, may require this file to be generated again.
 
-# Offense count: 172
+# Offense count: 154
 # This cop supports safe autocorrection (--autocorrect).
 # Configuration parameters: Max, AllowHeredoc, AllowURI, AllowQualifiedName, URISchemes, AllowRBSInlineAnnotation, AllowCopDirectives, AllowedPatterns, SplitStrings.
 # URISchemes: http, https
@@ -88,17 +88,17 @@ Metrics/AbcSize:
 Metrics/BlockLength:
   Max: 53
 
-# Offense count: 37
+# Offense count: 35
 # Configuration parameters: AllowedMethods, AllowedPatterns, Max.
 Metrics/CyclomaticComplexity:
   Enabled: false
 
-# Offense count: 68
+# Offense count: 74
 # Configuration parameters: CountComments, CountAsOne, AllowedMethods, AllowedPatterns.
 Metrics/MethodLength:
   Max: 62
 
-# Offense count: 29
+# Offense count: 26
 # Configuration parameters: AllowedMethods, AllowedPatterns, Max.
 Metrics/PerceivedComplexity:
   Enabled: false
@@ -168,7 +168,7 @@ RSpec/ContextWording:
 RSpec/DescribeClass:
   Enabled: false
 
-# Offense count: 654
+# Offense count: 652
 # Configuration parameters: CountAsOne.
 RSpec/ExampleLength:
   Max: 30
@@ -222,16 +222,16 @@ RSpec/MessageSpies:
     - 'spec/lutaml/uml_repository/presenters/diagram_presenter_spec.rb'
     - 'spec/lutaml/uml_repository/repository_spec.rb'
 
-# Offense count: 38
+# Offense count: 39
 RSpec/MultipleExpectations:
-  Max: 6
+  Max: 7
 
-# Offense count: 151
+# Offense count: 149
 # Configuration parameters: AllowSubject.
 RSpec/MultipleMemoizedHelpers:
   Max: 11
 
-# Offense count: 35
+# Offense count: 33
 # Configuration parameters: AllowedGroups.
 RSpec/NestedGroups:
   Max: 4

data/lib/lutaml/cli/enhanced_formatter.rb

@@ -125,7 +125,7 @@ module Lutaml
       # @param klass [Object] Class object to display
       # @param path_formatter [Proc] Formatter for package paths
       # @return [String] Enhanced class details
-      def self.format_class_details_enhanced(klass, path_formatter = nil) # rubocop:disable Metrics/AbcSize,Metrics/CyclomaticComplexity,Metrics/MethodLength,Metrics/PerceivedComplexity
+      def self.format_class_details_enhanced(klass, _path_formatter = nil) # rubocop:disable Metrics/AbcSize,Metrics/CyclomaticComplexity,Metrics/MethodLength,Metrics/PerceivedComplexity
         lines = []
 
         # Header box
@@ -137,25 +137,21 @@ module Lutaml
 
         # Basic information
         lines << colorize("Basic Information:", :cyan)
-        if klass.respond_to?(:xmi_id)
+        if klass.is_a?(Lutaml::Uml::TopElement) && klass.xmi_id
           lines << "  XMI ID:      #{klass.xmi_id}"
         end
-        if klass.respond_to?(:stereotype) && klass.stereotype && !klass.stereotype.empty?
+        if klass.is_a?(Lutaml::Uml::TopElement) && klass.stereotype && !klass.stereotype.empty?
           st = klass.stereotype
           st_str = st.is_a?(Array) ? st.join(", ") : st
           lines << "  Stereotype:  #{st_str}"
         end
-        if klass.respond_to?(:is_abstract)
+        if klass.is_a?(Lutaml::Uml::Classifier)
           lines << "  Abstract:    #{klass.is_abstract ? 'Yes' : 'No'}"
         end
-        if path_formatter && klass.respond_to?(:package)
-          package_path = path_formatter.call(klass.package)
-          lines << "  Package:     #{ICONS[:package]} #{package_path}"
-        end
         lines << ""
 
         # Attributes
-        if klass.respond_to?(:attributes) && klass.attributes &&
+        if klass.is_a?(Lutaml::Uml::Classifier) && klass.attributes &&
             !klass.attributes.empty?
           lines << colorize(
             "#{ICONS[:attribute]} Attributes (#{klass.attributes.size}):",
@@ -173,21 +169,21 @@ module Lutaml
         end
 
         # Operations
-        if klass.respond_to?(:operations) && klass.operations &&
+        if klass.is_a?(Lutaml::Uml::Classifier) && klass.operations &&
             !klass.operations.empty?
           lines << colorize(
             "#{ICONS[:operation]} Operations (#{klass.operations.size}):",
             :yellow,
           )
           klass.operations.each do |op|
-            params = if op.respond_to?(:parameters) && op.parameters
-                       op.parameters.map do |p|
+            params = if op.owned_parameter
+                       op.owned_parameter.map do |p|
                          "#{p.name}: #{p.type}"
                        end.join(", ")
                      else
                        ""
                      end
-            return_type = if op.respond_to?(:return_type) && op.return_type
+            return_type = if op.return_type
                             " : #{op.return_type}"
                           else
                             ""
@@ -486,18 +482,13 @@ module Lutaml
       #
       # @param attr [Object] Attribute with cardinality
       # @return [String] Formatted cardinality
-      def self.format_cardinality(attr) # rubocop:disable Metrics/CyclomaticComplexity,Metrics/PerceivedComplexity
-        return "" unless attr.respond_to?(:cardinality)
+      def self.format_cardinality(attr)
         return "" unless attr.cardinality
 
         card = attr.cardinality
-        if card.respond_to?(:min) && card.respond_to?(:max)
-          min = card.min || "0"
-          max = card.max || "*"
-          "[#{min}..#{max}]"
-        else
-          ""
-        end
+        min = card.min || "0"
+        max = card.max || "*"
+        "[#{min}..#{max}]"
       end
     end
   end

data/lib/lutaml/cli/interactive_shell/command_base.rb

@@ -14,7 +14,7 @@ module Lutaml
         def current_path = shell.current_path
 
         def current_path=(path)
-          shell.instance_variable_set(:@current_path, path)
+          shell.current_path = path
         end
 
         def config = shell.config
@@ -22,7 +22,7 @@ module Lutaml
         def last_results = shell.last_results
 
         def last_results=(results)
-          shell.instance_variable_set(:@last_results, results)
+          shell.last_results = results
         end
 
         def path_history = shell.path_history

data/lib/lutaml/cli/interactive_shell/query_commands.rb

@@ -99,7 +99,7 @@ module Lutaml
             puts ""
             puts "Name: #{cls.name}"
 
-            if cls.respond_to?(:attributes) && cls.attributes && !cls.attributes.empty?
+            if cls.is_a?(Lutaml::Uml::Classifier) && cls.attributes && !cls.attributes.empty?
               puts ""
               puts OutputFormatter.colorize("Attributes:", :yellow)
               cls.attributes.each do |attr|
@@ -110,7 +110,7 @@ module Lutaml
         end
 
         def show_package(path)
-          nav = shell.instance_variable_get(:@navigation)
+          nav = shell.navigation
           path = nav.resolve_path(path)
           pkg = repository.find_package(path)
 

data/lib/lutaml/cli/interactive_shell.rb

@@ -17,8 +17,11 @@ module Lutaml
       HISTORY_FILE = File.expand_path("~/.lutaml-xmi-history")
       MAX_HISTORY = 1000
 
-      attr_reader :repository, :current_path, :config, :bookmarks,
-                  :last_results, :path_history
+      attr_reader :repository, :config, :bookmarks,
+                  :path_history,
+                  :navigation, :query, :bookmarks_cmd, :export, :help
+
+      attr_accessor :current_path, :last_results
 
       def initialize(lur_path_or_repo, config: nil)
         @config = {
@@ -83,25 +86,25 @@ module Lutaml
 
       COMMAND_DISPATCH = {
         # Navigation
-        "cd" => :@navigation, "pwd" => :@navigation,
-        "ls" => :@navigation, "list" => :@navigation,
-        "tree" => :@navigation, "up" => :@navigation,
-        "root" => :@navigation, "back" => :@navigation,
+        "cd" => :navigation, "pwd" => :navigation,
+        "ls" => :navigation, "list" => :navigation,
+        "tree" => :navigation, "up" => :navigation,
+        "root" => :navigation, "back" => :navigation,
         # Query
-        "find" => :@query, "f" => :@query,
-        "show" => :@query, "s" => :@query,
-        "search" => :@query, "?" => :@query,
-        "results" => :@query,
+        "find" => :query, "f" => :query,
+        "show" => :query, "s" => :query,
+        "search" => :query, "?" => :query,
+        "results" => :query,
         # Bookmarks
-        "bookmark" => :@bookmarks_cmd, "bm" => :@bookmarks_cmd,
+        "bookmark" => :bookmarks_cmd, "bm" => :bookmarks_cmd,
         # Export
-        "export" => :@export,
+        "export" => :export,
         # Utilities
-        "help" => :@help, "h" => :@help,
-        "history" => :@help,
-        "clear" => :@help, "cls" => :@help,
-        "config" => :@help,
-        "stats" => :@help
+        "help" => :help, "h" => :help,
+        "history" => :help,
+        "clear" => :help, "cls" => :help,
+        "config" => :help,
+        "stats" => :help
       }.freeze
 
       METHOD_MAP = {
@@ -136,7 +139,7 @@ module Lutaml
         method_name = METHOD_MAP[command]
 
         if handler_var && method_name
-          instance_variable_get(handler_var).send(method_name, args)
+          public_send(handler_var).public_send(method_name, args)
         else
           puts OutputFormatter.warning("Unknown command: #{command}")
           puts "Type 'help' for available commands"

data/lib/lutaml/cli/output_formatter.rb

@@ -69,7 +69,7 @@ module Lutaml
           data
         else
           # For complex objects, convert to string representation
-          data.respond_to?(:name) ? data.name : data.to_s
+          data.is_a?(Lutaml::Uml::TopElement) ? data.name : data.to_s
         end
       end
 
@@ -271,7 +271,7 @@ module Lutaml
           stats[:most_complex_classes].first(5).each do |item|
             klass = item[:class]
             complexity = item[:complexity]
-            class_name = klass.respond_to?(:name) ? klass.name : klass.to_s
+            class_name = klass&.name || klass.to_s
 
             lines << "  #{class_name} (complexity: #{complexity})"
           end

data/lib/lutaml/cli/resource_registry.rb

@@ -10,7 +10,7 @@ module Lutaml
     #
     # @example Get configuration for a type
     #   config = ResourceRegistry.config_for(:class)
-    #   classes = repository.send(config[:list_method])
+    #   classes = repository.public_send(config[:list_method])
     #
     # @example List all available types
     #   ResourceRegistry.types # => [:package, :class, :diagram, ...]

data/lib/lutaml/cli/tree_view_formatter.rb

@@ -163,7 +163,7 @@ module Lutaml
         children = []
 
         # Add attributes
-        if @show_attributes && klass.respond_to?(:attributes) &&
+        if @show_attributes && klass.is_a?(Lutaml::Uml::Classifier) &&
             klass.attributes
           children.concat(klass.attributes.map do |a|
             { type: :attribute, obj: a }
@@ -171,7 +171,7 @@ module Lutaml
         end
 
         # Add operations
-        if @show_operations && klass.respond_to?(:operations) &&
+        if @show_operations && klass.is_a?(Lutaml::Uml::Classifier) &&
             klass.operations
           children.concat(klass.operations.map do |o|
             { type: :operation, obj: o }
@@ -239,11 +239,7 @@ module Lutaml
         return "" if @max_depth && depth >= @max_depth
 
         assoc_name = assoc.name || "(unnamed)"
-        target = if assoc.respond_to?(:member_end) && assoc.member_end&.first
-                   assoc.member_end.first.xmi_type || "Unknown"
-                 else
-                   "Unknown"
-                 end
+        target = assoc.member_end || "Unknown"
 
         connector = is_last ? TREE_CHARS[:last_branch] : TREE_CHARS[:branch]
         icon = ICONS[:association]
@@ -280,7 +276,7 @@ module Lutaml
       # Determine class type (class, interface, enumeration)
       def determine_class_type(klass)
         return :enumeration if klass.class.name&.include?("Enum")
-        return :interface if klass.respond_to?(:stereotype) &&
+        return :interface if klass.is_a?(Lutaml::Uml::TopElement) &&
           Array(klass.stereotype).any? { |s| s&.downcase == "interface" }
 
         :class

data/lib/lutaml/cli/uml/build_command.rb

@@ -141,7 +141,7 @@ module Lutaml
             OutputFormatter.progress_done
 
             # Display verbose validation if requested
-            if options[:verbose] && result.respond_to?(:validation_details)
+            if options[:verbose] && result.validation_details
               display_verbose_validation(result.validation_details)
             end
 
@@ -149,8 +149,7 @@ module Lutaml
               handle_validation_result(result)
 
               # Display unique unresolved types if present
-              if result.respond_to?(:external_references) &&
-                  result.external_references.any?
+              if result.external_references&.any?
                 display_unresolved_types(result.external_references)
               end
 

data/lib/lutaml/cli/uml/diagram_command.rb

@@ -83,12 +83,12 @@ module Lutaml
             }
 
             # Add stereotype if available
-            if uml_element.respond_to?(:stereotype) && uml_element.stereotype && !uml_element.stereotype.empty?
+            if uml_element.is_a?(Lutaml::Uml::TopElement) && uml_element.stereotype && !uml_element.stereotype.empty?
               element_data[:stereotype] = uml_element.stereotype.first
             end
 
             # Add attributes and operations for classes
-            if uml_element.respond_to?(:attributes) && uml_element.attributes
+            if uml_element.is_a?(Lutaml::Uml::Classifier) && uml_element.attributes
               element_data[:attributes] = uml_element.attributes.map do |attr|
                 {
                   name: attr.name,
@@ -98,7 +98,7 @@ module Lutaml
               end
             end
 
-            if uml_element.respond_to?(:operations) && uml_element.operations
+            if uml_element.is_a?(Lutaml::Uml::Classifier) && uml_element.operations
               element_data[:operations] = uml_element.operations.map do |op|
                 {
                   name: op.name,
@@ -131,30 +131,30 @@ module Lutaml
             }
 
             # Add role and multiplicity information if available
-            if connector.respond_to?(:owner_end_attribute_name) &&
+            if connector.is_a?(Lutaml::Uml::Association) &&
                 connector.owner_end_attribute_name
               connector_data[:source_role] = connector.owner_end_attribute_name
             end
 
-            if connector.respond_to?(:member_end_attribute_name) &&
+            if connector.is_a?(Lutaml::Uml::Association) &&
                 connector.member_end_attribute_name
               connector_data[:target_role] = connector.member_end_attribute_name
             end
 
-            if connector.respond_to?(:owner_end_cardinality) &&
+            if connector.is_a?(Lutaml::Uml::Association) &&
                 connector.owner_end_cardinality
               connector_data[:source_multiplicity] =
                 format_cardinality(connector.owner_end_cardinality)
             end
 
-            if connector.respond_to?(:member_end_cardinality) &&
+            if connector.is_a?(Lutaml::Uml::Association) &&
                 connector.member_end_cardinality
               connector_data[:target_multiplicity] =
                 format_cardinality(connector.member_end_cardinality)
             end
 
             # Add source and target information if available
-            if connector.respond_to?(:source) && connector.source
+            if connector.is_a?(Lutaml::Uml::Association) && connector.owner_end
               source_obj = find_diagram_object_for_element(
                 connector.source.xmi_id, diagram
               )
@@ -166,7 +166,7 @@ module Lutaml
               end
             end
 
-            if connector.respond_to?(:target) && connector.target
+            if connector.is_a?(Lutaml::Uml::Association) && connector.member_end
               target_obj = find_diagram_object_for_element(
                 connector.target.xmi_id, diagram
               )
@@ -405,11 +405,7 @@ module Lutaml
         def format_cardinality(cardinality)
           return "" unless cardinality
 
-          if cardinality.respond_to?(:to_s)
-            cardinality.to_s
-          else
-            ""
-          end
+          cardinality.to_s
         end
       end
     end

data/lib/lutaml/cli/uml/find_command.rb

@@ -84,7 +84,7 @@ module Lutaml
           end
 
           output = classes.map do |cls|
-            cls.respond_to?(:name) ? (cls.name || cls.to_s) : cls.to_s
+            cls.is_a?(Lutaml::Uml::TopElement) ? cls.name : cls.to_s
           end
 
           # output result based on the format option

data/lib/lutaml/cli/uml/inspect_command.rb

@@ -66,7 +66,7 @@ module Lutaml
           config = ResourceRegistry.config_for(identifier.type)
           return nil unless config
 
-          repo.send(config[:find_method], identifier.path)
+          repo.public_send(config[:find_method], identifier.path)
         end
 
         def display_element_details(element, identifier, repo)

data/lib/lutaml/converter/dsl_to_uml.rb

@@ -101,14 +101,14 @@ module Lutaml
               .split("\n").map(&:strip).join("\n")
           end
 
-          if model.respond_to?("#{key}=")
-            if model.class.attributes[key.to_sym].options[:collection]
-              values = model.send(key.to_sym).to_a
-              values << value
-              model.send("#{key}=", values)
-            else
-              model.send("#{key}=", value)
-            end
+          next unless model.class.attributes.key?(key.to_sym)
+
+          if model.class.attributes[key.to_sym].options[:collection]
+            values = model.public_send(key.to_sym).to_a
+            values << value
+            model.public_send("#{key}=", values)
+          else
+            model.public_send("#{key}=", value)
           end
         end
       end

data/lib/lutaml/converter/xmi_to_uml.rb

@@ -271,7 +271,7 @@ module Lutaml
           uml_type = operation.uml_type.first
           uml_type_idref = uml_type.idref if uml_type
 
-          if !operation.respond_to?(:association) || operation.association.nil?
+          if !operation.class.attributes.key?(:association) || operation.association.nil?
             ::Lutaml::Uml::Operation.new.tap do |op|
               op.id = operation.id
               op.xmi_id = uml_type_idref
@@ -289,7 +289,7 @@ module Lutaml
         # In ea-xmi-2.5.1, constraints are moved to source/target under
         # connectors
         constraints = %i[source target].map do |st|
-          connector_node.send(st).constraints.constraint
+          connector_node.public_send(st).constraints.constraint
         end.flatten
 
         constraints.map do |constraint|

data/lib/lutaml/ea/diagram/configuration.rb

@@ -47,14 +47,14 @@ module Lutaml
           value = nil
 
           # 1. Try class-specific override (highest priority)
-          if element.respond_to?(:name) && element.name
+          if element.is_a?(Lutaml::Uml::TopElement) && element.name
             class_config = dig_config("classes.#{element.name}.#{property}")
             value = class_config if class_config
           end
 
           # 2. Try package-based styling
           if !value &&
-              element.respond_to?(:package_name) && element.package_name
+              element.is_a?(Lutaml::Uml::Diagram) && element.package_name
             # Support wildcards: "CityGML::*"
             package_configs = config_data["packages"] || {}
             package_configs.each do |pattern, pkg_config|
@@ -67,7 +67,7 @@ module Lutaml
           end
 
           # 3. Try stereotype-based styling
-          if !value && element.respond_to?(:stereotype) && element.stereotype
+          if !value && element.is_a?(Lutaml::Uml::TopElement) && element.stereotype
             stereotypes = Array(element.stereotype)
             stereotypes.each do |stereo|
               stereo_value = dig_config("stereotypes.#{stereo}.#{property}")

data/lib/lutaml/ea/diagram/style_parser.rb

@@ -139,6 +139,21 @@ module Lutaml
           base_style
         end
 
+        # Convert EA color integer to hex color
+        # (EA stores colors as BGR integers)
+        # @param ea_color [Integer] EA color value
+        # @return [String] Hex color string
+        def color_from_ea_color(ea_color)
+          return EA_COLORS[:class_fill] if ea_color.zero?
+
+          # EA colors are stored as BGR, convert to RGB
+          b = (ea_color & 0xFF0000) >> 16
+          g = (ea_color & 0x00FF00) >> 8
+          r = ea_color & 0x0000FF
+
+          format("#%02x%02x%02x", r, g, b).upcase # rubocop:disable Style/FormatStringToken
+        end
+
         private
 
         def get_base_element_style(element_type) # rubocop:disable Metrics/AbcSize,Metrics/MethodLength
@@ -277,21 +292,6 @@ module Lutaml
           style
         end
 
-        # Convert EA color integer to hex color
-        # (EA stores colors as BGR integers)
-        # @param ea_color [Integer] EA color value
-        # @return [String] Hex color string
-        def color_from_ea_color(ea_color)
-          return EA_COLORS[:class_fill] if ea_color.zero?
-
-          # EA colors are stored as BGR, convert to RGB
-          b = (ea_color & 0xFF0000) >> 16
-          g = (ea_color & 0x00FF00) >> 8
-          r = ea_color & 0x0000FF
-
-          format("#%02x%02x%02x", r, g, b).upcase # rubocop:disable Style/FormatStringToken
-        end
-
         # Get stereotype-specific styling
         # (EA has specific colors for stereotypes)
         # @param stereotype [String] Stereotype name

data/lib/lutaml/ea/diagram/style_resolver.rb

@@ -182,11 +182,12 @@ module Lutaml
             case key.strip
             when "BCol"
               # Background color (BGR integer)
-              style[:fill] = style_parser.send(:color_from_ea_color, value.to_i)
+              style[:fill] =
+                style_parser.color_from_ea_color(value.to_i)
             when "LCol"
               # Line color (BGR integer)
               style[:stroke] =
-                style_parser.send(:color_from_ea_color, value.to_i)
+                style_parser.color_from_ea_color(value.to_i)
             when "BFol"
               # Bold font (0 or 1)
               style[:font_weight] = value == "1" ? 700 : 400
@@ -220,7 +221,7 @@ module Lutaml
             when "LCol"
               # Line color
               style[:stroke] =
-                style_parser.send(:color_from_ea_color, value.to_i)
+                style_parser.color_from_ea_color(value.to_i)
             when "LWth"
               # Line width
               style[:stroke_width] = value.to_i
@@ -263,23 +264,15 @@ module Lutaml
         #
         # @param connector [Object] Association connector
         # @return [String] Specific association type
-        def determine_association_type(connector) # rubocop:disable Metrics/CyclomaticComplexity,Metrics/MethodLength
-          return "association" unless connector.respond_to?(:member_end)
-          return "association" unless connector.member_end
-
-          # Ensure member_end is an array (handle legacy data
-          # where it might be a string)
-          member_ends = Array(connector.member_end)
-
-          # Check for aggregation or composition
-          member_ends.each do |end_point|
-            if end_point.respond_to?(:aggregation)
-              case end_point.aggregation&.downcase
-              when "shared"
-                return "aggregation"
-              when "composite"
-                return "composition"
-              end
+        def determine_association_type(connector)
+          return "association" unless connector.is_a?(Lutaml::Uml::Association)
+
+          [connector.owner_end_type, connector.member_end_type].each do |type|
+            case type&.downcase
+            when "aggregation"
+              return "aggregation"
+            when "composition"
+              return "composition"
             end
           end
 

data/lib/lutaml/formatter/graphviz.rb

@@ -237,7 +237,7 @@ module Lutaml
         HEREDOC
 
         field_table = format_member_rows(node.attributes, hide_members)
-        method_table = if node.respond_to?(:operations)
+        method_table = if node.operations&.any?
                          format_member_rows(node.operations, hide_members)
                        end
         table_body = [name_html, field_table, method_table].map do |type|