def test_request_subset_of_available_edge_features(self): params = ParameterServer() requested_features = GraphObserver.available_edge_attributes()[0:2] params["ML"]["GraphObserver"]["EnabledEdgeFeatures"] = requested_features observer = GraphObserver(params=params) self.assertEqual( observer._enabled_edge_attribute_keys, requested_features)
def test_request_partially_invalid_edge_features(self): params = ParameterServer() requested_features =\ GraphObserver.available_edge_attributes()[0:2] + ['invalid'] params["ML"]["GraphObserver"]["EnabledEdgeFeatures"] = requested_features observer = GraphObserver(params=params) # remove invalid feature from expected list requested_features.pop(-1) self.assertEqual( observer._enabled_edge_attribute_keys, requested_features)