--- a/json2yolosegment.xml	Fri Jun 13 11:23:35 2025 +0000
+++ b/json2yolosegment.xml	Mon Jul 07 06:47:08 2025 +0000
@@ -5,16 +5,24 @@
     </macros>
     <expand macro="creator" />
     <expand macro="edam" />
+    <expand macro="requirements" />
     <command detect_errors="aggressive">
     <![CDATA[
       mkdir ./input ./output &&
 
       #for $filename in $in_json:
-          ln -s '$filename' './input/${filename.element_identifier}' &&
+          #if $filename.element_identifier.endswith($filename.ext)
+              ln -s '$filename' './input/${filename.element_identifier}' &&
+	  #else:
+	      ln -s '$filename' './input/${filename.element_identifier}.${filename.ext}' &&
+          #end if
       #end for
-		    
-      python '$__tool_directory__/json2yolosegment.py' -i ./input/ -o ./output -c '$class_name'
-		
+
+      python '$__tool_directory__/json2yolosegment.py'
+	-i ./input/
+	-o ./output
+	-c '$class_name'
+	
     ]]>
     </command>
     <inputs>
@@ -45,6 +53,24 @@
                 </element>
             </output_collection>
         </test>
+        <test>
+            <param name="in_json" value="in_json_noext,in_json1_noext" />
+            <param name="class_name" value="class_names.txt" />
+	    <output_collection name="output_yolo">
+                <element name="in_json1_noext">
+                    <assert_contents>
+                        <has_n_lines n="1" />
+                        <has_text text="0.7710371819960861" />
+                    </assert_contents>
+                </element>
+                <element name="in_json_noext">
+                    <assert_contents>
+                        <has_n_lines n="1" />
+                        <has_text text="0.8532289628180039" />
+                    </assert_contents>
+                </element>
+            </output_collection>
+        </test>
     </tests>
     <help>
 <![CDATA[