/
kachernenko
/
aaa_tests
Обзор
Документация
Войти
/
kachernenko
/
aaa_tests
Код
Запросы
1
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
develop_branch
test_one_hot_encoder.py
45 строк
2 KB
KAChernenko
hw done
04 ноя 2024, 16:23
04 ноя 2024, 16:23
815620a
Код
Авторство
О чём код?
import unittest from one_hot_encoder import fit_transform class TestFitTransform(unittest.TestCase): def test_single_category(self): """Проверяем одно значение""" result = fit_transform("cat") expected = [("cat", [1])] self.assertEqual(result, expected) def test_multiple_categories(self): """Проверяем несколько различных значений""" result = fit_transform("cat", "dog", "bird") expected = [ ("cat", [0, 0, 1]), ("dog", [0, 1, 0]), ("bird", [1, 0, 0]) ] self.assertEqual(result, expected) def test_repeated_categories(self): """Проверяем повторяющиеся значения""" result = fit_transform("cat", "dog", "cat") expected = [ ("cat", [0, 1]), ("dog", [1, 0]), ("cat", [0, 1]) ] self.assertEqual(result, expected) def test_no_arguments(self): """Проверяем обработку исключения, когда аргументы не переданы""" with self.assertRaises(TypeError): fit_transform() def test_assert_not_in_result(self): """Проверяем, что несуществующая категория не попала в результат""" result = fit_transform("apple", "banana") categories = [item[0] for item in result] self.assertNotIn("orange", categories) if __name__ == '__main__': unittest.main()