class TestFormulaTask(object): """Test FormulaTask. """ def setup(self): self.root = RootTask(should_stop=Event(), should_pause=Event()) self.task = FormulaTask(name='Test') self.root.add_child_task(0, self.task) def test_perform1(self): """Test checking that the evaluated formula gets written to the database """ self.task.formulas = OrderedDict([('key1', "1.0+3.0"), ('key2', '3.0+4.0')]) self.root.prepare() self.task.perform() assert (self.task.get_from_database('Test_key1') == 4.0 and self.task.get_from_database('Test_key2') == 7.0) def test_perform_from_load(self): """Test checking for correct loading from pref and that we can still recall values from the database """ self.task.write_in_database('pi', 3.1) self.task.formulas = ordered_dict_from_pref(self, self.task.formulas, "[(u'key1', '1.0+3.0'), (u'key2', '3.0 + {Test_pi}')]") self.root.prepare() self.task.perform() assert (self.task.get_from_database('Test_key1') == 4.0 and self.task.get_from_database('Test_key2') == 6.1) def test_check(self): """Test checking that an unformattable formula gives an error """ self.task.formulas = OrderedDict([('key1', "1.0+3.0"), ('key2', '3.0+4.0 + {Test_pi}')]) test, traceback = self.task.check() assert not test assert len(traceback) == 1 assert 'root/Test-key2' in traceback
def test_view(windows): """Test the FormulaTask view. """ show_and_close_widget(FormulaView(task=FormulaTask(name='Test')))
def setup(self): self.root = RootTask(should_stop=Event(), should_pause=Event()) self.task = FormulaTask(name='Test') self.root.add_child_task(0, self.task)