Log In Sign Up
Models:
Daniel Galbraith
DanielG/
MediAug
0
Downloads: 1
[dd6877]: / tests / test_augment.py

Download this file

15 lines (7 with data), 285 Bytes 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
from unittest import TestCase



from mediaug.augment import Pipeline

from mediaug.dataset import Dataset





class TestAugment(TestCase):



    def test_pipeline_init(self):

        ds = Dataset('/Users/sean/projects/cancerDetection/data/cells')

        pipeline = Pipeline(ds)