def testState(self): state = dict(val=100) fs = FunctionSource(dataFunction, state=state, resetFieldName="reset", sequenceIdFieldName="sequence") self.assertIsNotNone(fs) r = fs.getNextRecordDict() self.assertIsNotNone(r) r = fs.getNextRecordDict() r = fs.getNextRecordDict() self.assertEqual(103, state["val"])
def testState(self): state = dict(val=100) fs = FunctionSource(dataFunction, state=state, resetFieldName="reset", sequenceIdFieldName="sequence") self.assertIsNotNone(fs) r = fs.getNextRecordDict() self.assertIsNotNone(r) r = fs.getNextRecordDict() r = fs.getNextRecordDict() self.assertEqual(103, state["val"])
def testResetAndSequenceFields(self): fs = FunctionSource(dataFunction, state=None, resetFieldName="reset", sequenceIdFieldName="sequence") self.assertIsNotNone(fs) r = fs.getNextRecordDict() self.assertIsNotNone(r)
def testDefaultArgs(self): fs = FunctionSource(dataFunction, state=None, resetFieldName=None, sequenceIdFieldName=None) self.assertIsNotNone(fs) r = fs.getNextRecordDict() self.assertIsNotNone(r)
def testPickle(self): state = dict(val=100) fs = FunctionSource(dataFunction, state=state, resetFieldName="reset", sequenceIdFieldName="sequence") self.assertIsNotNone(fs) r = fs.getNextRecordDict() self.assertIsNotNone(r) pkl = pickle.dumps(fs) self.assertIsNotNone(pkl) fs2 = pickle.loads(pkl) self.assertIsNotNone(fs2) r = fs2.getNextRecordDict() r = fs2.getNextRecordDict() self.assertEqual(103, fs2.state["val"])
def testPickle(self): state = dict(val=100) fs = FunctionSource(dataFunction, state=state, resetFieldName="reset", sequenceIdFieldName="sequence") self.assertIsNotNone(fs) r = fs.getNextRecordDict() self.assertIsNotNone(r) pkl = pickle.dumps(fs) self.assertIsNotNone(pkl) fs2 = pickle.loads(pkl) self.assertIsNotNone(fs2) r = fs2.getNextRecordDict() r = fs2.getNextRecordDict() self.assertEqual(103, fs2.state["val"])
def testResetAndSequenceFields(self): fs = FunctionSource(dataFunction, state=None, resetFieldName="reset", sequenceIdFieldName="sequence") self.assertIsNotNone(fs) r = fs.getNextRecordDict() self.assertIsNotNone(r)
def testDefaultArgs(self): fs = FunctionSource(dataFunction, state=None, resetFieldName=None, sequenceIdFieldName=None) self.assertIsNotNone(fs) r = fs.getNextRecordDict() self.assertIsNotNone(r)