def setUpClass(cls): config = FormatAnalyzer._load_train_config(merge_dicts( get_train_config(), { "javascript": { "feature_extractor": { "left_siblings_window": 1, "right_siblings_window": 1, "parents_depth": 1, "node_features": ["start_line", "reserved", "roles"], }, }, })) base = Path(__file__).parent with lzma.open(str(base / "benchmark.js.xz"), mode="rt") as fin: contents = fin.read() with lzma.open(str(base / "benchmark.uast.xz")) as fin: uast = bblfsh.Node.FromString(fin.read()) file = File(content=bytes(contents, "utf-8"), uast=uast) files = [file, file] cls.fe = FeatureExtractor(language="javascript", **config["javascript"]["feature_extractor"]) cls.fe.extract_features(files) cls.class_representations = cls.fe.composite_class_representations cls.n_classes = len(cls.fe.labels_to_class_sequences) cls.ordinal = cls.return_node_feature(FeatureId.start_line) cls.categorical = cls.return_node_feature(FeatureId.reserved) cls.bag = cls.return_node_feature(FeatureId.roles)
def setUpClass(cls): base = Path(__file__).parent # str() is needed for Python 3.5 with lzma.open(str(base / "benchmark.js.xz"), mode="rt") as fin: contents = fin.read() with lzma.open(str(base / "benchmark.uast.xz")) as fin: uast = bblfsh.Node.FromString(fin.read()) file = File(content=bytes(contents, "utf-8"), uast=uast) cls.files = [file] config = FormatAnalyzer._load_train_config(get_train_config()) cls.config = config["javascript"] cls.extractor = FeatureExtractor(language="javascript", **cls.config["feature_extractor"])
def setUpClass(cls): slogging_setup("DEBUG", False) cls.language = "javascript" cls.bblfsh_client = bblfsh.BblfshClient("0.0.0.0:9432") cls.data_service = FakeDataService(cls.bblfsh_client, files=None, changes=None) cls.stub = cls.data_service.get_bblfsh() cls.config = FormatAnalyzer._load_train_config({ "global": { "feature_extractor": { "cutoff_label_support": 0 } }, })[cls.language]["feature_extractor"]
def setUpClass(cls): cls.maxDiff = None base = Path(__file__).parent # str() is needed for Python 3.5 with lzma.open(str(base / "benchmark_small.js.xz"), mode="rt") as fin: contents = fin.read() with lzma.open(str(base / "benchmark_small.js.uast.xz")) as fin: uast = bblfsh.Node.FromString(fin.read()) config = FormatAnalyzer._load_train_config(get_train_config()) fe_config = config["javascript"] cls.feature_extractor = FeatureExtractor( language="javascript", label_composites=label_composites, **fe_config["feature_extractor"]) cls.file = File(content=bytes(contents, "utf-8"), uast=uast) cls.X, cls.y, (cls.vnodes_y, cls.vnodes, cls.vnode_parents, cls.node_parents) = \ cls.feature_extractor.extract_features([cls.file])
def setUp(self): config = FormatAnalyzer._load_train_config(get_train_config()) self.extractor = FeatureExtractor( language="javascript", **config["javascript"]["feature_extractor"])