Skip to content
Permalink
Browse files

fix(cli): fix cli chanel close

  • Loading branch information...
hanxiao committed Aug 30, 2019
1 parent d441f58 commit f1402f5038cfd8d2f6fff6fc4ce0ba783424d121
Showing with 2 additions and 1 deletion.
  1. +2 −1 tests/test_pipeline_train.py
@@ -22,6 +22,7 @@ def test_train(self):
self.assertRaises(RuntimeError, de.encode, 1)
de.train()
self.assertEqual(2, de.encode(1))
self.dirname = os.path.dirname(__file__)

def tearDown(self):
if os.path.exists('dummy-pipeline.bin'):
@@ -42,7 +43,7 @@ def test_pipeline_train(self):
self.assertEqual(4, a.encode(1))

def test_load_yaml(self):
p = BaseEncoder.load_yaml('yaml/pipeline-multi-encoder.yml')
p = BaseEncoder.load_yaml(os.path.join(self.dirname, 'yaml', 'pipeline-multi-encoder.yml'))
self.assertRaises(RuntimeError, p.encode, 1)
p.train(1)
self.assertEqual(5, p.encode(1))

0 comments on commit f1402f5

Please sign in to comment.
You can’t perform that action at this time.