Пример #1
0
    def test_is_available(self):
        di = DataInterpreter()
        di.requires = ['a', 'b']
            
        t1 = DummyTrial('t1', 1)
        t1.a = DummyResource('data')
        t1.b = DummyResource('data')

        t2 = DummyTrial('t2', 2)
        t2.a = DummyResource('data')
        t2.b = DummyResource(None)

        t3 = DummyTrial('t3', 3)
        t3.a = DummyResource('data')

        ok_trials = [t1]
        bad_trials_1 = [t2]
        bad_trials_2 = [t3]
        bad_trials_3 = [t2, t3]
        bad_trials_4 = [t1, t2, t3]

        self.assertEquals(di.is_available(ok_trials), True)
        self.assertEquals(di.is_available(bad_trials_1), False)
        self.assertEquals(di.is_available(bad_trials_2), False)
        self.assertEquals(di.is_available(bad_trials_3), False)
        self.assertEquals(di.is_available(bad_trials_4), False)
Пример #2
0
 def test_construct_filenames(self):
     di = DataInterpreter()
     di.name = 'Some Name'
     trials = [DummyTrial('t1', 1), DummyTrial('t2', 2)]
     base_path = 'BASEPATH'
     f1 = 't1__Some_Name'
     f2 = 't2__Some_Name'
     expected = {1:os.path.join(base_path, f1), 
                 2:os.path.join(base_path, f2)}
     self.assertEquals(di.construct_filenames(trials, base_path), expected)