From c8b917b1e0f217bf529918d540c49d07a69dba3e Mon Sep 17 00:00:00 2001
From: Szymon Szukalski <szymon@skas.io>
Date: Fri, 25 Oct 2024 01:51:28 +1100
Subject: Write test cases for new relationships

---
 lib/family_tree.rb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'lib')

diff --git a/lib/family_tree.rb b/lib/family_tree.rb
index 0c9ecef..0909ed1 100644
--- a/lib/family_tree.rb
+++ b/lib/family_tree.rb
@@ -83,7 +83,7 @@ class FamilyTree
 
     case relationship.downcase
     when 'child'
-      children.map(&:name).join(' ')
+      children.empty? ? 'NONE' : children.map(&:name).join(' ')
     when 'son'
       sons = children.select { |child| child.gender == Gender::MALE }
       sons.empty? ? 'NONE' : sons.map(&:name).join(' ')
-- 
cgit v1.2.3