Using Python Unittest to test results in an array

Question

I've implemented a function which calculate annual mortgage repayment, and I want to test whether it generates the correct results.

I've put input mortgage values in an array, such as:

input_array([   50000.,   100000.,   150000., ...])

and the corresponding results in another array:

output_array([   3200.60,   6401.20,   9601.79, ...])

I'd like to take each of the elements in the input array as the input of my function and test the result is the same as output_array.

Is it possible to do it automatically rather than input the mortgage value and expected output in the assertEqual function manually?

Many thanks.

poe123 · Accepted Answer · 2015-08-03 11:01:07Z

5

assertListEqual(map(f, input_array), output_array)

answered Aug 3, 2015 at 11:01

poe123

1,2189 silver badges11 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Ihor Pomaranskyy · Accepted Answer · 2015-08-03 11:04:27Z

4

You have two options. The first, just compare whole arrays:

expected_list = [1, 2, 3, 4]
output_list = calc_mortgage([10, 20, 30, 42])
self.assertEqual(expected_list, output_list)

The second, you can compare each element:

expected_list = [1, 2, 3, 4]
output_list = calc_mortgage([10, 20, 30, 42])
for pair in zip(expected_list, output_list):
    self.assertEqual(pair[0], pair[1])

answered Aug 3, 2015 at 11:04

Ihor Pomaranskyy

5,6691 gold badge36 silver badges41 bronze badges

Comments

robm · Accepted Answer · 2021-12-01 15:41:35Z

1

For numpy arrays, something like:

self.assertTrue(
    numpy.allclose(calc_mortgage([10, 20, 30, 42], [1, 2, 3, 4]),
    msg="calc_mortgage error"
)

answered Dec 1, 2021 at 15:41

robm

1,4031 gold badge18 silver badges25 bronze badges

Collectives™ on Stack Overflow

Using Python Unittest to test results in an array

3 Answers 3

Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related