
import unittest
import user_code
import ast
import re   
import importlib
import csv
import unittest
import importlib
import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt
import user_code

class TestTask(unittest.TestCase):
    def setUp(self):
        importlib.reload(user_code)
        self.wavelengths1 = [400, 450, 500, 550, 600, 650, 700]
        self.absorbance1 = [0.05, 0.15, 0.30, 0.25, 0.18, 0.10, 0.05]
        self.wavelengths2 = [400, 450, 500, 550, 600, 650, 700]
        self.absorbance2 = [0.08, 0.20, 0.28, 0.22, 0.16, 0.09, 0.04]
        plt.close('all')

    def test_two_lines_on_same_plot(self):
        plt.close('all')
        user_code.plot_two_absorbance_spectra(self.wavelengths1, self.absorbance1, self.wavelengths2, self.absorbance2)
        figs = [plt.figure(i) for i in plt.get_fignums()]
        _dynamic_test(self, len(figs) > 0, "A figure is created.", "No figure was created.")
        ax = figs[-1].axes[0] if figs[-1].axes else None
        _dynamic_test(self, ax is not None, "Axes found on figure.", "No axes found on the figure.")
        lines = ax.get_lines()
        _dynamic_test(self, len(lines) == 2, "Two lines are plotted on the same graph.", f"Expected 2 lines, found {len(lines)}.")

    def test_lines_have_different_colors(self):
        plt.close('all')
        user_code.plot_two_absorbance_spectra(self.wavelengths1, self.absorbance1, self.wavelengths2, self.absorbance2)
        figs = [plt.figure(i) for i in plt.get_fignums()]
        ax = figs[-1].axes[0] if figs[-1].axes else None
        lines = ax.get_lines() if ax else []
        colors = [line.get_color() for line in lines]
        _dynamic_test(self, len(colors) == 2 and colors[0] != colors[1], "Each line has a different color.", f"Expected two different colors, got: {colors}")

    def test_legend_present_and_labels(self):
        plt.close('all')
        user_code.plot_two_absorbance_spectra(self.wavelengths1, self.absorbance1, self.wavelengths2, self.absorbance2)
        figs = [plt.figure(i) for i in plt.get_fignums()]
        ax = figs[-1].axes[0] if figs[-1].axes else None
        legend = ax.get_legend() if ax else None
        _dynamic_test(self, legend is not None, "Legend is present on the plot.", "Legend is missing.")
        labels = [t.get_text() for t in legend.get_texts()] if legend else []
        _dynamic_test(self, len(labels) == 2 and all(label.strip() != '' for label in labels), "Legend has two labels for the samples.", f"Expected two legend labels, got: {labels}")

    def test_axis_labels_and_title(self):
        plt.close('all')
        user_code.plot_two_absorbance_spectra(self.wavelengths1, self.absorbance1, self.wavelengths2, self.absorbance2)
        figs = [plt.figure(i) for i in plt.get_fignums()]
        ax = figs[-1].axes[0] if figs[-1].axes else None
        xlabel = ax.get_xlabel() if ax else ''
        ylabel = ax.get_ylabel() if ax else ''
        title = ax.get_title() if ax else ''
        _dynamic_test(self, xlabel == 'Wavelength (nm)', "X-axis label is correct.", f"Expected x-axis label 'Wavelength (nm)', got '{xlabel}'")
        _dynamic_test(self, ylabel == 'Absorbance', "Y-axis label is correct.", f"Expected y-axis label 'Absorbance', got '{ylabel}'")
        _dynamic_test(self, title.strip() != '', "Title is present on the plot.", "Plot title is missing.")

def _dynamic_test(test_case, condition, success_message, failure_message):
    if condition:
        test_case._testMethodName = success_message
        test_case.assertTrue(True, success_message)
    else:
        test_case._testMethodName = failure_message
        test_case.fail(failure_message)

def normalize_text(text):
    text = text.lower()
    text = re.sub(r"\\s{2,}", " ", text)
    text = re.sub(r"\\s*([,:?])\\s*", r"\\1 ", text)
    return text.strip()

def change_var(code: str, var_name: str, value: str) -> str:
    tree = ast.parse(code)
    lines = code.splitlines()
    changed = False
    # Collect all assignment nodes to modify
    assign_nodes = [
        (i, node)
        for i, node in enumerate(tree.body)
        if isinstance(node, ast.Assign)
        and any(isinstance(target, ast.Name) and target.id == var_name for target in node.targets)
    ]

    # If nothing to change, return unmodified code
    if not assign_nodes:
        return code

    # Perform replacements for all matching assignments (from last to first to not break line offsets)
    for i, node in reversed(assign_nodes):
        start_line = node.lineno - 1
        line = lines[start_line]
        indent = ' ' * (len(line) - len(line.lstrip()))
        lines[start_line] = f"{indent}{var_name} = {value}"
        next_line = len(lines)
        for next_node in tree.body[i+1:]:
            if hasattr(next_node, 'lineno'):
                next_line = next_node.lineno - 1
                break
        if next_line > start_line + 1:
            lines[start_line+1:next_line] = []
        changed = True

    return '\\n'.join(lines) if changed else code

if __name__ == "__main__":
    unittest.main()


test_main.py

An engaging, hands-on course introducing Python programming through real-world chemistry problems. Learn to clean experimental data, perform molar calculations, visualize spectra, and analyze reaction rates using Python and scientific libraries.

Learn Python basics through chemistry-themed examples, focusing on variables, data types, and simple calculations relevant to laboratory work.

Dive into real-world chemistry data: clean, process, and analyze raw experimental results using Python.

Explore how to visualize and analyze chemical spectra and reaction rates using Python's scientific libraries.

Challenge: Compare Absorbance Spectra

解答