gettc 1.5 → 1.6

data/core/lib/{topcoder → gettc}/print.rb

@@ -1,7 +1,7 @@
-require 'topcoder/problem'
-require 'rdiscount'
+require "gettc/problem"
+require "rdiscount"
 
-module TopCoder
+module Gettc
     class Problem
         def to_md
             out  = "# [#{@name}](#{@url})\n"
@@ -10,19 +10,19 @@ module TopCoder
             out << "## Statement\n"
             out << @statement << "\n\n"
 
-            if not @definitions.empty? then
+            unless @definitions.empty? then
                 out << "## Definitions\n"
                 @definitions.each do |k, v| out << "- *#{k}*: `#{v}`\n" end 
                 out << "\n"
             end 
 
-            if not @notes.empty? then
+            unless @notes.empty? then
                 out << "## Notes\n"
                 @notes.each do |note| out << "- #{note}\n" end 
                 out << "\n"
             end 
 
-            if not @constraints.empty? then
+            unless @constraints.empty? then
                 out << "## Constraints\n"
                 @constraints.each do |constraint|
                     out << "- #{constraint}\n"
@@ -30,18 +30,18 @@ module TopCoder
                 out << "\n"
             end 
 
-            if not @examples.empty? then
+            unless @examples.empty? then
                 out << "## Examples\n"
                 @examples.each_index do |i|
                     example = @examples[i]
                     out << "### Example #{i + 1}\n"
                     out << "#### Input\n<c>"
-                    out << example.input.gsub("\n", '<br />')
+                    out << example.input.gsub("\n", "<br />")
                     out << "</c>\n"
                     out << "#### Output\n<c>"
-                    out << example.output.gsub("\n", '<br />')
+                    out << example.output.gsub("\n", "<br />")
                     out << "</c>\n"
-                    if not example.reason.empty? then
+                    unless example.reason.empty? then
                         out << "#### Reason\n#{example.reason}\n\n"
                     end 
                 end
@@ -53,4 +53,4 @@ module TopCoder
             return markdown.to_html
         end
     end
-end
+end

data/core/lib/{topcoder → gettc}/problem.rb

@@ -1,27 +1,27 @@
-module TopCoder
+module Gettc
     class Case
         attr_accessor :input, :output, :reason
         def initialize
-            @input = ''
-            @output = ''
-            @reason = ''
+            @input = ""
+            @output = ""
+            @reason = ""
         end
     end
     class Image
         attr_accessor :name, :content
         def initialize
-            @name = ''
-            @content = ''
+            @name = ""
+            @content = ""
         end
     end
     class Problem
         attr_accessor :name, :url, :source, :statement, :definitions, :notes, 
                       :constraints, :examples, :systests, :images
         def initialize
-            @name = ''
-            @url = ''
-            @source = ''
-            @statement = ''
+            @name = ""
+            @url = ""
+            @source = ""
+            @statement = ""
             @definitions = { }
             @notes = []
             @constraints = []
@@ -30,4 +30,4 @@ module TopCoder
             @images = []
         end       
     end
-end
+end

data/core/lib/{topcoder → gettc}/signature.rb

@@ -1,18 +1,18 @@
-require 'topcoder/types'
+require "gettc/types"
     
-module TopCoder 
+module Gettc
     class SignatureError < StandardError
     end
     class CannotParseSignature < SignatureError
         attr_accessor :source
-        def initialize source, msg = 'Cannot parse signature'
+        def initialize source, msg = "Cannot parse signature"
             @source = source
             super "#{msg} (#{source}"
         end
     end
     class InvalidVariableName < SignatureError
         attr_accessor :name
-        def initialize name, msg = 'Invalid variable name'
+        def initialize name, msg = "Invalid variable name"
             @name = name
             super "#{msg} (#{name})"            
         end
@@ -39,17 +39,17 @@ module TopCoder
     def parse_method_signature str
         str.strip!        
         sigs = []
-        parts = str.split '('
+        parts = str.split "("
         raise CannotParseSignature.new str if parts.size != 2
         sigs << parse_signature(parts[0])
 
         str = parts[1]
-        raise CannotParseSignature.new str if str[-1] != ')'
+        raise CannotParseSignature.new str if str[-1] != ")"
         str.chop!      
 
-        parts = str.split ','
+        parts = str.split ","
         parts.each do |sig| sigs << parse_signature(sig) end
 
         return sigs
     end
-end
+end

data/core/lib/{topcoder → gettc}/types.rb

@@ -1,11 +1,8 @@
-require 'singleton'
-    
-module TopCoder 
-    class TypeError < StandardError        
-    end
+module Gettc
+    TypeError = Class.new StandardError        
     class UnrecognizedType < TypeError
         attr_accessor :type
-        def initialize type = nil, msg = 'Not a valid TopCoder type'
+        def initialize type = nil, msg = "Not a valid TopCoder type"
             @type = type
             super "#{msg} (#{type})"         
         end
@@ -17,6 +14,26 @@ module TopCoder
         def obj?
             return @is_object
         end
+        def to_s
+            if self == TInt then
+                return "int"
+            elsif self == TLong then
+                return "long"        
+            elsif self == TFloat then
+                return "float"
+            elsif self == TDouble then
+                return "double"
+            elsif self == TChar then
+                return "char"
+            elsif self == TString then
+                return "String"
+            elsif self == TBoolean then
+                return "boolean"
+            elsif is_a? TArray then
+                return subtype.to_s + "[]"
+            end
+            return "unknown"
+        end
     end
     TBoolean = Type.new false
     TInt = Type.new false
@@ -28,11 +45,11 @@ module TopCoder
     class TArray < Type
         attr_accessor :subtype
         def initialize subtype
-            raise UnrecognizedType.new subtype if not subtype.is_a? Type
+            raise UnrecognizedType.new subtype unless subtype.is_a? Type
             @subtype = subtype
         end
         def == ary
-            return false if not ary.is_a? TArray
+            return false unless ary.is_a? TArray
             return @subtype == ary.subtype
         end
         def obj?
@@ -40,23 +57,23 @@ module TopCoder
         end
     end
     def parse_type str
-        return TArray.new parse_type str[0 .. -3] if str[-2 .. -1] == '[]'
+        return TArray.new parse_type str[0 .. -3] if str[-2 .. -1] == "[]"
         case str
-        when 'boolean'
+        when "boolean"
             return TBoolean
-        when 'int'
+        when "int"
             return TInt
-        when 'long'
+        when "long"
             return TLong
-        when 'float'
+        when "float"
             return TFloat
-        when 'double'
+        when "double"
             return TDouble
-        when 'char'
+        when "char"
             return TChar
-        when 'String'
+        when "String"
             return TString
         end
         raise UnrecognizedType.new str
     end
-end
+end

data/core/test/{topcoder → gettc}/download_test.rb

@@ -1,14 +1,14 @@
-require 'test/unit' 
-require 'topcoder/download' 
+require "test/unit" 
+require "topcoder/download" 
 include TopCoder
 
 class DownloadTest < Test::Unit::TestCase
     def setup
-        @account = Account.new 'gettc', 'algorithm'
+        @account = Account.new "gettc", "algorithm"
     end
     def test_wrong_account
         assert_raises LoginFailed do 
-            Downloader.new Account.new 'username', 'password'
+            Downloader.new Account.new "username", "password"
         end
     end    
     def test_wrong_id
@@ -23,7 +23,7 @@ class DownloadTest < Test::Unit::TestCase
         3.times do
             id = ids[rand ids.size]
             html = downloader.download_problem id
-            assert_match '<h3>Problem Statement</h3>', html
+            assert_match "<h3>Problem Statement</h3>", html
         end
     end
-end
+end

data/core/test/gettc/generate_test.rb

@@ -0,0 +1,31 @@
+require "test/unit" 
+require "fileutils"
+require "tmpdir"
+
+require "topcoder/generate" 
+include TopCoder
+
+class GenerateTest < Test::Unit::TestCase
+    def setup
+        @source_d = File.join File.dirname(__FILE__), "../../dist/template"
+        @target_d = File.join Dir.tmpdir, "gettc"
+        FileUtils.mkdir @target_d unless File.directory? @target_d
+        @generator = Generator.new @source_d, @target_d
+    end
+    def test_initialize
+        assert_raise SourceDirNotExist do 
+            Generator.new "this_directory_must_not_exist", @target_d
+        end
+        assert_raise TargetDirNotExist do 
+            Generator.new @source_d, "this_directory_must_not_exist"
+        end
+    end
+    def test_prob_dir_exists
+        prob = Problem.new
+        prob.name = "JustATest"
+        prob_d = File.join @target_d, prob.name
+        FileUtils.mkdir prob_d unless File.exists? prob_d
+        assert_raise ProblemDirExists do @generator.generate prob end
+        FileUtils.rmdir prob_d
+    end
+end

data/core/test/{topcoder → gettc}/parse_test.rb

@@ -1,34 +1,34 @@
-require 'test/unit' 
-require 'topcoder/parse' 
+require "test/unit" 
+require "topcoder/parse" 
 include TopCoder
 
 class ParseTest < Test::Unit::TestCase
     def setup
-        downloader = Downloader.new Account.new 'gettc', 'algorithm'
+        downloader = Downloader.new Account.new "gettc", "algorithm"
         @parser = Parser.new downloader
-        @datadir = File.join File.dirname(__FILE__), '../../files/stats'
+        @data_d = File.join File.dirname(__FILE__), "../../temp/download_problem_statement"
     end
     def get_problem_raw prob
-        return File.read File.join @datadir, prob + '.htm'
+        return File.read File.join @data_d, prob + ".htm"
     end
     def test_indexes
-        assert_equal [0, 3], @parser.indexes('abcde', 'bc')
-        assert_equal nil, @parser.indexes('abcde', 'f')
+        assert_equal [0, 3], @parser.indexes("abcde", "bc")
+        assert_equal nil, @parser.indexes("abcde", "f")
     end
     def test_filter
-        assert_equal 'Lorem Ipsum', @parser.filter('<xml>   Lorem Ipsum  </xml> ')
-        assert_equal '2^(3)', @parser.filter('2<sup>3</sup>')
-        assert_equal '*hi*', @parser.filter('   <b>hi</b>')
+        assert_equal "Lorem Ipsum", @parser.filter("<xml>   Lorem Ipsum  </xml> ")
+        assert_equal "2^(3)", @parser.filter("2<sup>3</sup>")
+        assert_equal "*hi*", @parser.filter("   <b>hi</b>")
         html = <<-END
             <img src=
             "http://www.topcoder.com/contest/problem/CirclesCountry/case1.gif">
         END
-        assert_equal '![image](images/case1.gif)', @parser.filter(html)
+        assert_equal "![image](images/case1.gif)", @parser.filter(html)
     end
     def test_PageNumbers
-        html = get_problem_raw 'PageNumbers'
+        html = get_problem_raw "PageNumbers"
         prob = @parser.parse html
-        assert_equal 'PageNumbers', prob.name
+        assert_equal "PageNumbers", prob.name
         assert_equal 5, prob.definitions.size
         assert_equal 1, prob.notes.size
         assert_equal 1, prob.constraints.size
@@ -37,9 +37,9 @@ class ParseTest < Test::Unit::TestCase
         assert_equal 0, prob.images.size
     end
     def test_CirclesCountry
-        html = get_problem_raw 'CirclesCountry'
+        html = get_problem_raw "CirclesCountry"
         prob = @parser.parse html
-        assert_equal 'CirclesCountry', prob.name
+        assert_equal "CirclesCountry", prob.name
         assert_equal 5, prob.definitions.size
         assert_equal 0, prob.notes.size
         assert_equal 7, prob.constraints.size
@@ -48,9 +48,9 @@ class ParseTest < Test::Unit::TestCase
         assert_equal 4, prob.images.size
     end
     def test_TheTournamentDivOne
-        html = get_problem_raw 'TheTournamentDivOne'
+        html = get_problem_raw "TheTournamentDivOne"
         prob = @parser.parse html
-        assert_equal 'TheTournamentDivOne', prob.name
+        assert_equal "TheTournamentDivOne", prob.name
         assert_equal 5, prob.definitions.size
         assert_equal 0, prob.notes.size
         assert_equal 4, prob.constraints.size
@@ -59,9 +59,9 @@ class ParseTest < Test::Unit::TestCase
         assert_equal 0, prob.images.size
     end
     def test_FunnyGames
-        html = get_problem_raw 'FunnyGames'
+        html = get_problem_raw "FunnyGames"
         prob = @parser.parse html
-        assert_equal 'FunnyGames', prob.name
+        assert_equal "FunnyGames", prob.name
         assert_equal 5, prob.definitions.size
         assert_equal 0, prob.notes.size
         assert_equal 6, prob.constraints.size
@@ -70,9 +70,9 @@ class ParseTest < Test::Unit::TestCase
         assert_equal 0, prob.images.size
     end
     def test_BuildingRoads
-        html = get_problem_raw 'BuildingRoads'
+        html = get_problem_raw "BuildingRoads"
         prob = @parser.parse html
-        assert_equal 'BuildingRoads', prob.name
+        assert_equal "BuildingRoads", prob.name
         assert_equal 5, prob.definitions.size
         assert_equal 2, prob.notes.size
         assert_equal 10, prob.constraints.size
@@ -81,9 +81,9 @@ class ParseTest < Test::Unit::TestCase
         assert_equal 2, prob.images.size
     end
     def test_Acronyms
-        html = get_problem_raw 'Acronyms'
+        html = get_problem_raw "Acronyms"
         prob = @parser.parse html
-        assert_equal 'Acronyms', prob.name
+        assert_equal "Acronyms", prob.name
         assert_equal 5, prob.definitions.size
         assert_equal 1, prob.notes.size
         assert_equal 8, prob.constraints.size
@@ -92,13 +92,13 @@ class ParseTest < Test::Unit::TestCase
         assert_equal 0, prob.images.size
     end
     def test_BackyardTrees
-        html = get_problem_raw 'BackyardTrees'
+        html = get_problem_raw "BackyardTrees"
         prob = @parser.parse html
-        assert_equal 'BackyardTrees', prob.name
+        assert_equal "BackyardTrees", prob.name
         assert_equal 5, prob.definitions.size
         assert_equal 0, prob.notes.size
         assert_equal 4, prob.constraints.size
         assert_equal 6, prob.examples.size
         assert_equal 2, prob.images.size
     end
-end
+end

data/core/test/gettc/signature_test.rb

@@ -0,0 +1,54 @@
+require "test/unit"
+require "topcoder/signature"
+include TopCoder
+
+class SignatureTest < Test::Unit::TestCase
+    def test_parse_signature
+        assert_raise CannotParseSignature do 
+            parse_signature "invalid_signature" 
+        end
+        assert_raise UnrecognizedType do 
+            parse_signature "strange_type name" 
+        end
+        assert_raise InvalidVariableName do 
+            parse_signature "int not&ok&name" 
+        end
+        assert_raise InvalidVariableName do 
+            parse_signature "int 0zero" 
+        end
+        sig = parse_signature "String[] a_valid_nam3"
+        assert_equal TArray.new(TString), sig.type
+        assert_equal "a_valid_nam3", sig.name
+    end
+    def test_parse_method_signature
+        assert_raise CannotParseSignature do
+            parse_method_signature "there are no brackets"
+        end
+        assert_raise CannotParseSignature do
+            parse_method_signature "int main(int main())"
+        end
+        assert_raise CannotParseSignature do
+            parse_method_signature "int main(oops forget to close bracket"
+        end
+        method  =  "  int   leastBorders(String[] X  , int[] Y, double[] R,"
+        method += "  char   my_x1,   long   y1  ,  float x2, int[][] y2) "
+        sigs = parse_method_signature method
+        assert_equal 8, sigs.size
+        assert_equal TInt, sigs[0].type
+        assert_equal "leastBorders", sigs[0].name
+        assert_equal TArray.new(TString), sigs[1].type
+        assert_equal "X", sigs[1].name
+        assert_equal TArray.new(TInt), sigs[2].type
+        assert_equal "Y", sigs[2].name
+        assert_equal TArray.new(TDouble), sigs[3].type
+        assert_equal "R", sigs[3].name
+        assert_equal TChar, sigs[4].type
+        assert_equal "my_x1", sigs[4].name
+        assert_equal TLong, sigs[5].type
+        assert_equal "y1", sigs[5].name
+        assert_equal TFloat, sigs[6].type
+        assert_equal "x2", sigs[6].name
+        assert_equal TArray.new(TArray.new(TInt)), sigs[7].type
+        assert_equal "y2", sigs[7].name
+    end
+end