def test_to_data_frame():
test = ContentAnalysisModel()
test.add_file(file_name="file1", label='file1', content='test')
test.add_file(file_name="file2", label='file2', content='other file')
test.add_dictionary(file_name="dict1.txt", label="dict1", content="test")
test.add_dictionary(file_name="dict2.txt", label="dict2", content="test")
test.count()
test.test_option = TestOptions(formula="")
test.save_formula()
test.generate_scores()
test.generate_averages()
assert isinstance(test.to_data_frame(), type(pd.DataFrame()))
def test_generate_averages():
test = ContentAnalysisModel()
test.add_file(file_name="file1", label='file1', content='test')
test.add_dictionary(file_name="dict1.txt", label="dict1", content="test")
test.count()
test.test_option = TestOptions(formula="0")
test.save_formula()
test.generate_scores()
test.generate_averages()
assert test.averages == [1.0, 0.0, 1.0, 0.0]
test = ContentAnalysisModel()
test.add_file(file_name="file1", label='file1', content='test')
test.add_file(file_name="file2", label='file2', content='other file')
test.add_dictionary(file_name="dict1.txt", label="dict1", content="test")
test.count()
test.test_option = TestOptions(formula="0")
test.save_formula()
test.generate_scores()
test.generate_averages()
assert test.averages == [0.5, 0.0, 1.5, 0.0]
test.count()
test.test_option = TestOptions(formula="4*[dict1]**2")
test.save_formula()
test.generate_scores()
test.generate_averages()
assert test.averages == [0.5, 2.0, 1.5, 2.0]
