a | b/tests/test_augment.py | ||
---|---|---|---|
1 | from unittest import TestCase |
||
2 | |||
3 | from mediaug.augment import Pipeline |
||
4 | from mediaug.dataset import Dataset |
||
5 | |||
6 | |||
7 | class TestAugment(TestCase): |
||
8 | |||
9 | def test_pipeline_init(self): |
||
10 | ds = Dataset('/Users/sean/projects/cancerDetection/data/cells') |
||
11 | pipeline = Pipeline(ds) |
||
12 | |||
13 | |||
14 | |||
15 |