示例#1
0
    def test_write_string(self):
        d = OrderedDict([("the", "a"), ("order", "z"), ("is", "b"), ("important", "y")])
        string = yaml.dump(d)

        loaded = yaml.load(string)
        assert type(loaded) == OrderedDict
        assert loaded.keys() == ["the", "order", "is", "important"]
示例#2
0
    def test_write_ordereddict(self):
        d = OrderedDict([("the", "a"), ("order", "z"), ("is", "b"), ("important", "y")])
        yaml.dump(d, open(os.path.join(self.dir_path, "output.yaml"), "w"))

        loaded = yaml.load(open(os.path.join(self.dir_path, "output.yaml"), "r"))
        assert type(loaded) == OrderedDict
        assert loaded.keys() == ["the", "order", "is", "important"]
示例#3
0
 def test_load_string(self):
     loaded = yaml.load("""
     the: a
     order: z
     of: b
     the_: y
     keys: c
     is: x
     important: d
     """)
     assert type(loaded) == OrderedDict
     assert loaded.keys() == ["the", "order", "of", "the_", "keys", "is", "important"]
示例#4
0
 def test_load_ordereddict(self):
     open(os.path.join(self.dir_path, "input.yaml"), "w").write("""
     the: a
     order: z
     of: b
     the_: y
     keys: c
     is: x
     important: d
     """)
     loaded = yaml.load(open(os.path.join(self.dir_path, "input.yaml"), "r"))
     assert type(loaded) == OrderedDict
     assert loaded.keys() == ["the", "order", "of", "the_", "keys", "is", "important"]
示例#5
0
    test_dir_number = 0
    while(True):
        if not os.path.exists(os.path.join(args.testdir, str(test_dir_number))):
            os.mkdir(os.path.join(args.testdir, str(test_dir_number)))
            break
        else:
            test_dir_number += 1

    test_dir = os.path.join(args.testdir, str(test_dir_number))

    for folder in ["input", "output", "tests", ".internal", ".internal/input"]:
        os.mkdir(os.path.join(test_dir, folder))

    problem_ids = {}
    try:
        data = yaml.load(open(os.path.join(args.task, 'task.yaml')))["problems"]
        for key, value in data.iteritems():
            if value["type"] == "code" and "boxes" in value:
                for box in value["boxes"]:
                    if value["boxes"][box]["type"] not in ["text", "file"]:
                        problem_ids[key+"/"+box] = False
                    elif value["boxes"][box]["type"] == "file":
                        problem_ids[key + "/" + box] = True
            elif value["type"] == "code-file":
                problem_ids[key] = True
            else:
                problem_ids[key] = False
    except:
        print "Cannot read task.yaml!"
        exit(1)