def setUp(self): self.hklcalc = Mock() self.hklcalc.constraints.report_constraints_lines.return_value = ['report1', 'report2'] self.commands = YouHklCommands(self.hklcalc)
class TestYouHklCommands: def setUp(self): self.hklcalc = Mock() self.hklcalc.constraints.report_constraints_lines.return_value = ['report1', 'report2'] self.commands = YouHklCommands(self.hklcalc) # def test__str__(self): # self.hklcalc.__str__.return_value = 'abc' # eq_(self.commands.__str__(), 'abc') def test_con_with_1_constraint(self): self.commands.con('cona') self.hklcalc.constraints.constrain.assert_called_with('cona') def test_con_with_1_constraint_with_value(self): self.commands.con('cona', 123) self.hklcalc.constraints.constrain.assert_called_with('cona') self.hklcalc.constraints.set_constraint.assert_called_with('cona', 123) def test_con_with_3_constraints(self): self.commands.con('cona', 'conb', 'conc') self.hklcalc.constraints.clear_constraints.assert_called() calls = [call('cona'), call('conb'), call('conc')] self.hklcalc.constraints.constrain.assert_has_calls(calls) def test_con_with_3_constraints_first_val(self): self.commands.con('cona', 1, 'conb', 'conc') self.hklcalc.constraints.clear_constraints.assert_called() calls = [call('cona'), call('conb'), call('conc')] self.hklcalc.constraints.constrain.assert_has_calls(calls) self.hklcalc.constraints.set_constraint.assert_called_with('cona', 1) def test_con_with_3_constraints_second_val(self): self.commands.con('cona', 'conb', 2, 'conc') self.hklcalc.constraints.clear_constraints.assert_called() calls = [call('cona'), call('conb'), call('conc')] self.hklcalc.constraints.constrain.assert_has_calls(calls) self.hklcalc.constraints.set_constraint.assert_called_with('conb', 2) def test_con_with_3_constraints_third_val(self): self.commands.con('cona', 'conb', 'conc', 3) self.hklcalc.constraints.clear_constraints.assert_called() calls = [call('cona'), call('conb'), call('conc')] self.hklcalc.constraints.constrain.assert_has_calls(calls) self.hklcalc.constraints.set_constraint.assert_called_with('conc', 3) def test_con_with_3_constraints_all_vals(self): self.commands.con('cona', 1, 'conb', 2, 'conc', 3) self.hklcalc.constraints.clear_constraints.assert_called() calls = [call('cona'), call('conb'), call('conc')] self.hklcalc.constraints.constrain.assert_has_calls(calls) calls = [call('cona', 1), call('conb', 2), call('conc', 3)] self.hklcalc.constraints.set_constraint.assert_has_calls(calls) def test_con_messages_and_help_visually(self): self.commands.con() print "**" print self.commands.con.__doc__ def test_con_message_display_whenn_selecting_an_unimplmented_mode(self): self.hklcalc.constraints.is_fully_constrained.return_value = True self.hklcalc.constraints.is_current_mode_implemented.return_value = False self.commands.con('phi', 'chi', 'eta')