Updated some unit tests

                   
|Q            |A  |
|---          |---|
|Bug Fix?     |n  |
|New Feature? |n  |
|BC Breaks?   |n  |
|Deprecations?|n  |
|Tests Pass?  |y  |
|Fixed Tickets|   |
|License      |MIT|
|Doc PR       |   |
                   

Updated unit tests[D[D[D