> ## Documentation Index
> Fetch the complete documentation index at: https://demo.agenticlabs.com/llms.txt
> Use this file to discover all available pages before exploring further.

# resolve_umi_sequence_test.py

Here's a detailed documentation of the target file `test/preprocess_tests/resolve_umi_sequence_test.py`:

## High-level description

This file contains unit tests for the UMI Resolution module in the Cassiopeia preprocessing pipeline. It tests the functionality of resolving UMI sequences and filtering cells based on UMI and read count thresholds.

## Code Structure

The main class `TestResolveUMISequence` inherits from `unittest.TestCase` and contains several test methods. It uses a sample collapsed UMI table to test the `resolve_umi_sequence` function from the `pipeline` module.

## Symbols

### `TestResolveUMISequence`

#### Description

A test case class that contains methods to test the UMI resolution functionality.

#### Internal Logic

1. Sets up a sample collapsed UMI table in the `setUp` method.
2. Contains test methods to verify different aspects of UMI resolution.

### `setUp`

#### Description

Initializes the test environment by creating a sample collapsed UMI table and setting up a temporary directory.

### `test_resolve_umi`

#### Description

Tests the basic functionality of the `resolve_umi_sequence` function.

#### Internal Logic

1. Calls `resolve_umi_sequence` with the sample data.
2. Checks if the correct sequence was selected for cell1-UMIA.
3. Verifies that cell2 was filtered out.
4. Ensures cell3 retained both UMIs.
5. Checks the expected read counts for each cell.

### `test_filter_by_reads`

#### Description

Tests the filtering of cells based on the average number of reads per UMI.

#### Internal Logic

1. Calls `resolve_umi_sequence` with a higher `min_avg_reads_per_umi` threshold.
2. Verifies that only the expected cells (cell3) remain after filtering.
3. Checks that the expected removed cells (cell1, cell2) are not in the result.

### `tearDown`

#### Description

Cleans up the temporary directory after each test.

## Dependencies

* `unittest`: Python's built-in unit testing framework
* `os`: For file path operations
* `shutil`: For directory removal
* `tempfile`: For creating temporary directories
* `pandas`: For data manipulation
* `cassiopeia.preprocess.pipeline`: The module being tested

## Configuration

The tests use a predefined collapsed UMI table and various parameters for the `resolve_umi_sequence` function.

## Error Handling

The tests use assertions to verify the expected outcomes. If any assertion fails, the test will raise an `AssertionError`.

## Notes

* The tests use a small, predefined dataset to verify the functionality of the UMI resolution process.
* The `plot` parameter is set to `False` in `test_resolve_umi` and `True` in `test_filter_by_reads`, demonstrating testing with and without plot generation.
* The temporary directory is used to store any output files generated during the tests and is cleaned up after each test.

This test file ensures that the UMI resolution functionality in the Cassiopeia preprocessing pipeline works as expected, particularly in handling different cell and UMI filtering scenarios.