class TestValidator(unittest.TestCase): def setUp(self): fields = ['f1', 'f2'] self.v = FieldExistsValidator(fields) def test_validate(self): assert self.v.validate({'f1': 'v1', 'f2': 'v2'}) assert not self.v.validate({'f1': 'v1'}) def test_execute(self): e1 = {'f1': 'v1', 'f2': 'v2'} e2 = {'f1': 'v1'} rdd = peachbox.Spark.Instance().context().parallelize([e1, e2]) p = self.v.execute(rdd).collect() self.assertEqual(1, len(p)) self.assertEqual('v1', p[0]['f1'])
class TestValidator(unittest.TestCase): def setUp(self): fields = ['f1', 'f2'] self.v = FieldExistsValidator(fields) def test_validate(self): assert self.v.validate({'f1':'v1', 'f2':'v2'}) assert not self.v.validate({'f1':'v1'}) def test_execute(self): e1 = {'f1':'v1', 'f2':'v2'} e2 = {'f1':'v1'} rdd = peachbox.Spark.Instance().context().parallelize([e1, e2]) p = self.v.execute(rdd).collect() self.assertEqual(1, len(p)) self.assertEqual('v1', p[0]['f1'])
def setUp(self): fields = ['f1', 'f2'] self.v = FieldExistsValidator(fields)