Ejemplo n.º 1
0
 def test_len(self):
     actual = src.fit_transform(['Moscow', 'New York', 'Moscow', 'London'])
     expected = [
         ('Moscow', [0, 0, 1]),
         ('New York', [0, 1, 0]),
         ('Moscow', [0, 0, 1]),
         ('London', [1, 0, 0]),
     ]
     self.assertTrue(len(actual) == len(expected))
Ejemplo n.º 2
0
def test_len():
    actual = fit_transform(['Moscow', 'New York', 'Moscow', 'London'])
    expected = [
        ('Moscow', [0, 0, 1]),
        ('New York', [0, 1, 0]),
        ('Moscow', [0, 0, 1]),
        ('London', [1, 0, 0]),
    ]
    assert (len(actual) == len(expected))
Ejemplo n.º 3
0
    def test_tf(self):
        actual = src.fit_transform(['Moscow', 'New York', 'Moscow', 'London'])
        expected = [
            ('Moscow', [0, 0, 1]),
            ('New York', [0, 1, 0]),
            ('Moscow', [0, 0, 1]),
            ('London', [1, 0, 0]),
        ]

        self.assertEqual(actual, expected)
Ejemplo n.º 4
0
 def test_empty(self):
     actual = src.fit_transform([])
     expected = []
     self.assertEqual(actual, expected)
Ejemplo n.º 5
0
def test_empty():
    actual = fit_transform([])
    expected = []
    assert (actual == expected)