def test_count_stuff():
"""
Tests that the top function-flow combination for the component "screw" is "couple solid"
"""
script_dir = os.path.dirname(__file__)
file1 = os.path.join(script_dir, '../assets/bladeCombined.csv')
comb_sort = count_stuff(file1)
assert comb_sort['screw'][0][0] == 'couple solid'
def test_get_top_results():
"""
Tests that the top 70% of function-flow combinations for the component "screw" only has one result
"""
script_dir = os.path.dirname(__file__)
file2 = os.path.join(script_dir, '../assets/bladeCombined.csv')
comb_sort = count_stuff(file2)
threshold = 0.7
thresh_results = get_top_results(comb_sort, threshold)
assert len(thresh_results['screw']) == 1
def test_1():
""" Example showing how to automate result finding with probability values """
# Dataset used for data mining
script_dir = os.path.dirname(__file__)
file1 = os.path.join(script_dir, '../assets/bladeCombined.csv')
comb_sort = count_stuff(file1)
# Use a threshold to get the top XX% of confidence values
threshold = 0.5
thresh_results = get_top_results(comb_sort, threshold)
# Use a known product for verification
input_file = os.path.join(script_dir, '../assets/InputExample.csv')
# Get dictionary of functions and flows for each component based on data mining
results, unmatched = get_func_rep(thresh_results, input_file, True)
assert results['screw'][0][0] == 'couple solid'
assert 'cheese' in unmatched
from autofunc.simple_counter import count_stuff from autofunc.get_func_rep import get_func_rep from autofunc.get_top_results import get_top_results from autofunc.write_results import write_results_from_dict import os.path """ Example showing how to automate functional representation using simple counting """ # Dataset used for data mining script_dir = os.path.dirname(__file__) file1 = os.path.join(script_dir, '../assets/bladeCombined.csv') comb_sort = count_stuff(file1) # Use a threshold to get the top XX% of confidence values threshold = 0.5 thresh_results = get_top_results(comb_sort, threshold) # Use a known product for verification input_file = os.path.join(script_dir, '../assets/InputExample.csv') # Get dictionary of functions and flows for each component based on data mining results, unmatched = get_func_rep(thresh_results, input_file, True) # Optional write to file - uncomment and rename to write file write_results_from_dict(results, 'test1.csv')