def setUp(self): from simple import StatsTable self.stats = StatsTable(CORPUS)
class TestStatsTable(unittest.TestCase): def setUp(self): from simple import StatsTable self.stats = StatsTable(CORPUS) def test_counts(self): self.assertEqual(1, self.stats.count("quick")) self.assertEqual(2, self.stats.count("the")) self.assertEqual(2, self.stats.count("The")) self.assertEqual(2.0 / 9, self.stats.frequency("the")) self.assertAlmostEqual(0.1102, self.stats.stddev(), places=4) def test_add(self): self.assertEqual(1, self.stats.count("quick")) self.stats.add("quick") self.assertEqual(2, self.stats.count("quick")) self.stats.add("quick") self.assertEqual(3, self.stats.count("quick")) def test_reset(self): self.assertEqual(1, self.stats.count("quick")) self.stats.add("quick") self.assertEqual(2, self.stats.count("quick")) self.stats.reset(CORPUS) self.assertEqual(1, self.stats.count("quick"))
class TestStatsTable(unittest.TestCase): def setUp(self): from simple import StatsTable self.stats = StatsTable(CORPUS) def test_counts(self): self.assertEqual(1, self.stats.count('quick')) self.assertEqual(2, self.stats.count('the')) self.assertEqual(2, self.stats.count('The')) self.assertEqual(2.0 / 9, self.stats.frequency('the')) self.assertAlmostEqual(.1102, self.stats.stddev(), places=4) def test_add(self): self.assertEqual(1, self.stats.count('quick')) self.stats.add('quick') self.assertEqual(2, self.stats.count('quick')) self.stats.add('quick') self.assertEqual(3, self.stats.count('quick')) def test_reset(self): self.assertEqual(1, self.stats.count('quick')) self.stats.add('quick') self.assertEqual(2, self.stats.count('quick')) self.stats.reset(CORPUS) self.assertEqual(1, self.stats.count('quick'))