def test_token_distribution_example_1(): """ Test with example 1. :return: """ distribution = get_token_distribution("abc 123") assert_true("abc" in distribution) assert_equal(distribution["abc"], 1) assert_true("123" in distribution) assert_equal(distribution["123"], 1)
def test_token_distribution_example_1(self): distribution = get_token_distribution("abc 123") self.assertTrue("abc" in distribution) self.assertEqual(1, distribution["abc"]) self.assertTrue("123" in distribution) self.assertEqual(1, distribution["123"])