Exemplo n.º 1
0
    def setUp(self):
        self.log.info("Setting up the engine")

        template_path = pjoin(self.test_context.engine_directory,
                              "recommendation-engine")
        engine_json_path = pjoin(self.test_context.data_directory,
                                 "quickstart_test/engine.json")

        self.training_data_path = pjoin(self.test_context.data_directory,
                                        "quickstart_test/training_data.txt")

        # downloading training data
        srun('curl https://raw.githubusercontent.com/apache/spark/master/' \
                'data/mllib/sample_movielens_data.txt --create-dirs -o {}'
                .format(self.training_data_path))

        app_context = AppContext(name="MyRecommender",
                                 template=template_path,
                                 engine_json_path=engine_json_path)

        self.app = AppEngine(self.test_context, app_context)
 def setUp(self):
     template_path = pjoin(self.test_context.engine_directory,
                           "recommendation-engine")
     app_context = AppContext(name="MyRecommender", template=template_path)
     self.app = AppEngine(self.test_context, app_context)