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))
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))
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)
def test_empty(self): actual = src.fit_transform([]) expected = [] self.assertEqual(actual, expected)
def test_empty(): actual = fit_transform([]) expected = [] assert (actual == expected)