def testIgnoresComment(self): self.assertDictEqual({}, ycsb._ParseWorkload('#\n')) self.assertDictEqual({}, ycsb._ParseWorkload('#recordcount = 10\n' '# columnfamily=cf')) self.assertDictEqual({'recordcount': '10'}, ycsb._ParseWorkload('#Sample!\nrecordcount = 10'))
def testParsesSampleWorkload(self): test_file_path = os.path.join(os.path.dirname(__file__), '..', 'data', 'ycsb_workloada') with open(test_file_path) as fp: contents = fp.read() actual = ycsb._ParseWorkload(contents) expected = { 'recordcount': '1000', 'operationcount': '1000', 'workload': 'com.yahoo.ycsb.workloads.CoreWorkload', 'readallfields': 'true', 'readproportion': '0.5', 'updateproportion': '0.5', 'scanproportion': '0', 'insertproportion': '0', 'requestdistribution': 'zipfian' } self.assertDictEqual(expected, actual)
def testParsesEmptyString(self): self.assertDictEqual({}, ycsb._ParseWorkload(''))