lib/take/unit/generator.rb in take-0.0.6 vs lib/take/unit/generator.rb in take-0.0.7

@@ -14,11 +14,11 @@
           walk(@parent)
           @output
         end
       end
 
-      def write_file
-        File.open("#{@options[:directory]}#{@parent.name}.c", "w") do |f|
+      def write_file(output)
+        File.open(output, "w") do |f|
           f.write generate
         end
       end
 
       def walk(node, *opts)