# Path: peachbox/task/import_properties.py
# class ImportProperties(Task):
#     """
#     Import attributes to define an edge.
#     An edge (or also called `relationship`) relates two nodes (also called entity) with each other.
#     Abstract class. Edged must implement functions: `rh_side`, `lh_side` and `partition_key`.
#     """
#     
# 
#     def __init__(self):
#         """Model defines schema of edge and must inhert from `peachbox.model.MasterDataSet`."""
#         self.ms = None
# 
#     def set_master_schema(self, master_schema):
#         self.ms = master_schema
# 
#     def execute(self, rdd):
#         if not self.ms:
#             raise AttributeError
#         return self.ms.fill_properties(rdd)


import unittest
import peachbox
from peachbox.task import ImportProperties


class MockMasterSchema(object):
    def fill_properties(self, rdd):
        return rdd
        

class TestImportProperties(unittest.TestCase):

    def test_init(self):
        imp = ImportProperties()
        assert imp


    def test_execute(self):

# Path: peachbox/task/import_properties.py
# class ImportProperties(Task):
#     """
#     Import attributes to define an edge.
#     An edge (or also called `relationship`) relates two nodes (also called entity) with each other.
#     Abstract class. Edged must implement functions: `rh_side`, `lh_side` and `partition_key`.
#     """
#     
# 
#     def __init__(self):
#         """Model defines schema of edge and must inhert from `peachbox.model.MasterDataSet`."""
#         self.ms = None
# 
#     def set_master_schema(self, master_schema):
#         self.ms = master_schema
# 
#     def execute(self, rdd):
#         if not self.ms:
#             raise AttributeError
#         return self.ms.fill_properties(rdd)


# Path: peachbox/task/tests/test_import_properties.py
import unittest
import peachbox
from peachbox.task import ImportProperties



class MockMasterSchema(object):
    def fill_properties(self, rdd):
        return rdd
        

class TestImportProperties(unittest.TestCase):

    def test_init(self):
        imp = ImportProperties()
        assert imp


    def test_execute(self):

        imp = ImportProperties()