Skip to content

siRNA featurizer bug? #8

Description

When adding new tests in tests/test_featurizer_kmer.py, transforming the siRNA featurizer raises an error. _extract_features_test in oligogym/features.py might be missing RNA strands and getting wrong k-mer counts


@pytest.mark.parametrize("featurizer", [kmer_merged()])
def test_feature_extraction_aso(featurizer):
    features = featurizer.fit_transform([ASO_TEST])
    features_sub = featurizer.transform([ASO_TEST])
    assert features.to_dict() == ASO_TEST_KMER_DICT
    assert features_sub.to_dict() == ASO_TEST_KMER_DICT


@pytest.mark.parametrize("featurizer", [kmer_merged(), kmer_split()])
def test_feature_extraction_sirna(featurizer):
    features = featurizer.fit_transform([SIRNA_TEST])
    if featurizer.split_strands:
        assert features.to_dict() == SIRNA_TEST_KMER_SPLIT_DICT
    else:
        assert features.to_dict() == SIRNA_TEST_KMER_MERGED_DICT

    features_sub = featurizer.transform([SIRNA_TEST])
    if featurizer.split_strands:
        assert features_sub.to_dict() == SIRNA_TEST_KMER_SPLIT_DICT
    else:
        assert features_sub.to_dict() == SIRNA_TEST_KMER_MERGED_DICT

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions