def test_main(self, createTarget, floodWeights, createSets, getShaders, pm ): blendMock = MagicMock(name='blendShapeMock') pm.blendShape.side_effect = blendMock n = MagicMock(name='neutralMock') n.name.return_value = 'name' n.numVertices.return_value = 2 t = MagicMock(name='targetMock') SeparateBlends.main(neutral=n, target=t) pm.mel.eval.assert_called_with("select -r name.vtx[0:1]") floodWeights.assert_called_with(blendNode=blendMock().__getitem__(),value=0) assert createSets.called assert getShaders.called assert createTarget.called assert pm.sceneName.called