add more fit tests

tests/test_baselines.py

@@ -6,6 +6,7 @@
 from moses.baselines import CombinatorialGenerator
 from moses.baselines import NGram
 from moses.baselines import HMM
+from moses.metrics.utils import fragmenter
 
 
 class test_baselines(unittest.TestCase):
@@ -28,6 +29,17 @@ def test_hmm(self):
         )
 
     def test_combinatorial(self):
+        self.assertEqual(
+            self.train[0],
+            'CCCS(=O)c1ccc2[nH]c(=NC(=O)OC)[nH]c2c1',
+            "Train set is different: %s" % self.train[0]
+        )
+        fragments = fragmenter('CCCS(=O)c1ccc2[nH]c(=NC(=O)OC)[nH]c2c1')
+        self.assertEqual(
+            fragments,
+            ['[1*]C(=O)N=c1[nH]c2ccc(S(=O)CCC)cc2[nH]1', '[3*]OC'],
+            "Fragmenter is not working properly: %s" % str(fragments)
+        )
         model = CombinatorialGenerator()
         model.fit(self.train[:100])
         self.assertEqual(