class TestProjectConfig(unittest.TestCase): def setUp(self): project_path = get_file_dir(__file__) config_file = project_path + '/master.yaml' self.master = AIFlowMaster(config_file=config_file) self.master.start() def tearDown(self): self.master.stop() self.master._clear_db() def test_load_project_config(self): project_path = get_file_dir(__file__) set_project_config_file(project_path + "/project.yaml") project_desc = get_project_description_from(project_path) self.assertEqual(project_desc.project_config.get_master_uri(), "localhost:50051") self.assertIsNone( project_desc.project_config.get('ai_flow config', None)) self.assertEqual(project_desc.project_config['ai_flow_home'], '/opt/ai_flow') self.assertEqual( project_desc.project_config['ai_flow_job_master.host'], 'localhost') self.assertEqual( project_desc.project_config['ai_flow_job_master.port'], 8081) self.assertEqual(project_desc.project_config['ai_flow_conf'], 'taskmanager.slot=2')
model_meta = af.ModelMeta(name="test", model_type="saved_model") model_version = af.ModelVersionMeta(version="11111", model_path="./tmp/saved_model/", model_metric="./tmp/saved_model/", model_id=0) processed = af.predict( input_data_list=[input_example], model_info=model_meta, model_version_info=model_version, executor=faf.flink_executor.FlinkJavaExecutor( java_class="com.apache.flink.ai.flow.TestPredict")) af.write_example(input_data=processed, example_info=example_2, exec_args=ExecuteArgs( batch_properties=batch_args_2, stream_properties=stream_args_2)) g = af.default_graph() workflow = af.compile_workflow(project_path=test_util.get_project_path()) print(dumps(list(workflow.jobs.values())[0])) if __name__ == '__main__': config_file = test_util.get_master_config_file() master = AIFlowMaster(config_file=config_file) master.start() test_util.set_project_config(__file__) run_flink_predict_job() master.stop()
def test_master_start_stop(self): config = MasterConfig() config.set_db_uri(db_type=DBType.SQLITE, uri="sqlite:///sql.db") master = AIFlowMaster(config_file=test_util.get_master_config_file()) master.start(is_block=False) master.stop()