import json from kwik import kwik import sys if __name__== "__main__": if (not len(sys.argv) == 2): raise ValueError("Must pass a valid json test case") with open(sys.argv[1]) as test_file: data = json.load(test_file) results = kwik(data) result_str = "" for r in results: result_str = result_str + ',' + r print result_str
b = np.array(a) episodes = map(list, b[:, 0]) answers = list(b[:,1]) num = len(episodes[0]) result = { 'name': 'student_example{0}'.format(ex), 'numOfPatrons': num, 'atEstablishment': episodes, 'fightOccurred': answers, 'output': output, } pp.pprint(result) games = data() for game in games: name = "test_{0}".format(game['name']) result = kwik(game) t = t_generator(game['output'], result) setattr(TestFight, name, t) maxIdks = game['numOfPatrons'] * (game['numOfPatrons'] - 1) - 1 name = "test_{0}_maxidks".format(game['name']) idks = len([idk for idk in result if idk == "I DON'T KNOW"]) t = t_max_generator(maxIdks, idks) setattr(TestFight, name, t) if __name__ == "__main__": unittest.main()