Python fit_transform Examples

Example #1

File: test_fit_transform_pytest.py Project: yarkaktus/avito_python_hw

def test_simple_case():
    cities = ["Moscow", "New York", "Moscow", "London"]
    exp_transformed_cities = [
        ("Moscow", [0, 0, 1]),
        ("New York", [0, 1, 0]),
        ("Moscow", [0, 0, 1]),
        ("London", [1, 0, 0]),
    ]
    assert fit_transform(cities) == exp_transformed_cities

Example #2

 def test_simple_case(self):
     cities = ["Moscow", "New York", "Moscow", "London"]
     exp_transformed_cities = [
         ("Moscow", [0, 0, 1]),
         ("New York", [0, 1, 0]),
         ("Moscow", [0, 0, 1]),
         ("London", [1, 0, 0]),
     ]
     self.assertEqual(fit_transform(cities), exp_transformed_cities)

Example #3

File: test_fit_transform_pytest.py Project: yarkaktus/avito_python_hw

def test_many_one_city():
    cities = ["Moscow", "Moscow", "Moscow", "Moscow", "Moscow"]
    exp_transformed_cities = [
        ("Moscow", [1]),
        ("Moscow", [1]),
        ("Moscow", [1]),
        ("Moscow", [1]),
        ("Moscow", [1]),
    ]
    assert fit_transform(cities) == exp_transformed_cities

Example #4

 def test_many_one_city(self):
     cities = ["Moscow", "Moscow", "Moscow", "Moscow", "Moscow"]
     exp_transformed_cities = [
         ("Moscow", [1]),
         ("Moscow", [1]),
         ("Moscow", [1]),
         ("Moscow", [1]),
         ("Moscow", [1]),
     ]
     self.assertEqual(fit_transform(cities), exp_transformed_cities)

Example #5

File: test_fit_transform_pytest.py Project: yarkaktus/avito_python_hw

def test_empty_args():
    with pytest.raises(TypeError):
        fit_transform()

Example #6

File: test_fit_transform_pytest.py Project: yarkaktus/avito_python_hw

def test_nontstr_args():
    with pytest.raises(TypeError):
        fit_transform(1, 2, 3)

Example #7

File: test_fit_transform_pytest.py Project: yarkaktus/avito_python_hw

def test_empty_list():
    assert fit_transform([]) == []

Example #8

 def test_empty_args(self):
     with self.assertRaises(TypeError):
         fit_transform()

Example #9

 def test_nontstr_args(self):
     with self.assertRaises(TypeError):
         fit_transform(1, 2, 3)

Example #10

 def test_empty_list(self):
     self.assertEqual(fit_transform([]), [])