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]
def test_generate_scores(): test = ContentAnalysisModel(TestOptions(formula="")) test.add_file(file_name="file1", label='file1', content='test') test.add_dictionary(file_name="dict1.txt", label="dict1", content="test") test.count() test.generate_scores() assert test.scores[0] == 0.0 test.test_option = TestOptions(formula="[dict1]") test.save_formula() test.generate_scores() assert test.scores[0] == 1 test = ContentAnalysisModel() test.add_file(file_name="file1", label='file1', content='test a') test.add_dictionary(file_name="dict1.txt", label="dict1", content="test") test.count() test.test_option = TestOptions(formula="[dict1]") test.save_formula() test.generate_scores() assert test.scores[0] == 0.5 test = ContentAnalysisModel() test.add_file(file_name="file1", label='file1', content='a test') test.add_dictionary(file_name="dict1.txt", label="dict1", content="test") test.count() test.test_option = TestOptions(formula="[dict1]*2") test.save_formula() test.generate_scores() assert test.scores[0] == 1 test = ContentAnalysisModel() test.add_file(file_name="file1", label='file1', content='a test a') test.add_dictionary(file_name="dict1.txt", label="dict1", content="test") test.count() test.test_option = TestOptions(formula="[dict1]") test.save_formula() test.generate_scores() assert test.scores[0] == round(1 / 3, 3)