class TestFunctions(unittest.TestCase): def setUp(self): self.capture = DataCapture(TEST_FILE) def testSetAndGet(self): a = range(10) b = range(20) self.capture.setData([a,b]) x, y = self.capture.getData() self.assertTrue(a == x) self.assertTrue(b == y)
def getCapture(filename): dc = DataCapture(filename) return dc.getData()