проверить, что список содержит те же типы «объектов» python unittest

Question

У меня есть тест, который проверяет возвращаемое значение функции, проблема в том, что функция возвращает список кортежей, где каждый кортеж содержит целое число и объект ,

Объект имеет атрибуты с плавающей запятой, поэтому я хочу использовать assertAlmostEqual, но что-то похожее на «assertSameStructure» в списке.

Таким образом, в псевдо-коде, что я хочу сделать:

import unittest 
class Test(unittest.TestCase): 
    def test_foo(self): 
     result = foo(data) 
     expected_obj = Bar(23.3,12.1) 
     self.assertSameStructure([0,expected_obj],result) 
     self.assertAlmostEquals(expected_obj,result[0][1]) 

Answer 1

Если вы уверены, что вы получите только один результат кортеж обратно из теста можно просто проверить

self.assertAlmostEquals([(0,expected_obj)], result) 

единственное, что вам не хватает в вашем псевдокоде, это скобки вокруг целого и объекта.

в качестве альтернативы вы можете реализовать assertSameStructure, запустив isinstance(obj, tuple) и т.д.

Answer 2

Есть несколько модулей, которые помогут проверить схемы, а не просто тестирования для определенного поведения. Вот простой пример с schema:

from schema import Schema 

class Test(unittest.TestCase): 
    def assertHasSchema(self, schema, obj): 
     return Schema(*schema).validate(obj) 

    def test_foo(self): 
     result = foo(data) 
     expected_obj = Bar(23.3,12.1) 

     self.assertHasSchema([(int, Bar)], result) 

     self.assertAlmostEquals(expected_obj, result[0][1])