Private/New-AZPDiskResponse.ps1

function New-AZPDiskResponse {
    param(
        [parameter(Mandatory)]
        [string] $cost,
        [parameter(Mandatory)]
        [string] $skuName,
        [parameter(Mandatory)]
        [string] $location,
        [parameter()]
        [ValidateSet('day','month')]
        [string] $unitOfMeasure = 'day'
    )

    $o = New-Object psobject
    $o | Add-Member -MemberType NoteProperty -Name "armSkuName" -Value $skuName
    if ($unitOfMeasure -eq 'day') {  
        $o | Add-Member -MemberType NoteProperty -Name "unitOfMeasure" -Value "1 day"
    } elseif (($unitOfMeasure -eq 'month')) {
        $o | Add-Member -MemberType NoteProperty -Name "unitOfMeasure" -Value "1 month"
    }
    $o | Add-Member -MemberType NoteProperty -Name "retailPrice" -Value $cost
    $o | Add-Member -MemberType NoteProperty -Name "location" -Value $location

    return $o
}