def test_query_filter(self):
flag_1 = analysis.isNBA("Will this team make it to the finals?")
flag_0 = analysis.isNBA("Michael Jordan is good!")
flag_n = analysis.isNBA("This is a random query...")
self.assertEqual(flag_1, 1)
self.assertEqual(flag_0, 0)
self.assertEqual(flag_n, -1)
def response(self):
"""
Function to return the the response bases on the query.
This function checks the query passed in by the user, and returns the
response from the appopriate query. The query should have already been
passed in when creating the "InferenceNetwork" object.
----------
self : none
Returns
-------
response : string
The appropriate response to the user's input query
"""
if self.node_type == "info":
node = InfoNode()
else:
# Check if query is NBA related
flag = isNBA(self.query)
if flag == 0:
return unsure
elif flag == -1:
return non_nba
# Query is definitely NBA related
if self.node_type == "rank":
node = RankNode()
if self.node_type == "stat":
node = StatNode()
node.load_query(self.query)
return node.response()
def process(self):
flag = isNBA(self.text)
if flag == -1:
return non_nba
elif flag == 0:
return unsure
else:
network = InferenceNetwork(self.text)
return network.response()
def response(self):
if self.node_type == "info":
node = InfoNode()
else:
# Check if query is NBA related
flag = isNBA(self.query)
if flag == 0:
return unsure
elif flag == -1:
return non_nba
# Query is definitely NBA related
if self.node_type == "rank":
node = RankNode()
if self.node_type == "stat":
node = StatNode()
node.load_query(self.query)
return node.response()
