def test_calculate_plot_variables(self): profile = MagicMock(selected_variables=["a", "b", "c"]) self.assertEqual(_calculate_plot_variables(profile, ["b", "c"]), ["b", "c"])
def test_calculate_plot_variables_3(self): profile = MagicMock(selected_variables=["a", "b", "c"]) # prints warning self.assertEqual(_calculate_plot_variables(profile, ["a", "d"]), ["a", "b", "c"])