added some more interesting RLE/BWT+RLE tests

chrislit · May 9, 2015 · 66646c8 · 66646c8
1 parent cd6aa4c
commit 66646c8
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/tests/test_distance.py b/tests/test_distance.py
@@ -1329,6 +1329,13 @@ def test_dist_compression_rle(self):
         self.assertAlmostEqual(dist_compression('abc', 'abc', 'bwtrle'), 0)
         self.assertAlmostEqual(dist_compression('abc', 'def', 'bwtrle'), 0.75)
 
+        self.assertAlmostEqual(dist_compression('aaa', 'bbaaa', 'rle'), 0.5)
+        self.assertAlmostEqual(dist_compression('abb', 'bbba', 'rle'), 1/3)
+        self.assertAlmostEqual(dist_compression('banana', 'banane', 'bwtrle'),
+                               0.57142857142)
+        self.assertAlmostEqual(dist_compression('bananas', 'bananen', 'bwtrle'),
+                               0.5)
+
     def test_sim_compression(self):
         """test abydos.distance.sim_compression
         """
@@ -1391,6 +1398,13 @@ def test_sim_compression_rle(self):
         self.assertAlmostEqual(sim_compression('abc', 'abc', 'bwtrle'), 1)
         self.assertAlmostEqual(sim_compression('abc', 'def', 'bwtrle'), 0.25)
 
+        self.assertAlmostEqual(sim_compression('aaa', 'bbaaa', 'rle'), 0.5)
+        self.assertAlmostEqual(sim_compression('abb', 'bbba', 'rle'), 2/3)
+        self.assertAlmostEqual(sim_compression('banana', 'banane', 'bwtrle'),
+                               0.42857142857)
+        self.assertAlmostEqual(sim_compression('bananas', 'bananen', 'bwtrle'),
+                               0.5)
+
 
 class MongeElkanTestCases(unittest.TestCase):
     """test cases for abydos.distance.sim_monge_elkan &