def METHOD_NAME(self, mock_vhdx):
    DiskFormat.new('vhdx', self.xml_state, 'root_dir', 'target_dir')
    mock_vhdx.assert_called_once_with(
        self.xml_state, 'root_dir', 'target_dir', {}
    )