def setUp(self): self.model = TreeModel(['Plot Type', 'Depth'], parent=None) self.model.load( os.getcwd() + '/radpy/plugins/BeamAnalysis/view/RFB/Unit Tests/Test1.rfb', 5, ['Plot Type', 'Depth'])