def test_rank_family(images):
result = remove_unidentified(images, rank="family", reset_index=True)
expected = pd.DataFrame({
"class": ["Aves", "Mammalia", "Mammalia"],
"order": ["Piciformes", "Carnivora", "Pilosa"],
"family": ["Picidae", "Felidae", "Myrmecophagidae"],
"genus": [np.nan, "Puma", "Myrmecophaga"],
"species": [np.nan, np.nan, "tridactyla"],
})
pd.testing.assert_frame_equal(result, expected)
def test_rank_epithet(images):
result = remove_unidentified(images, rank="epithet", reset_index=True)
expected = pd.DataFrame({
"class": ["Mammalia"],
"order": ["Pilosa"],
"family": ["Myrmecophagidae"],
"genus": ["Myrmecophaga"],
"species": ["tridactyla"],
})
pd.testing.assert_frame_equal(result, expected)
def test_intact_input(images):
images_original = images.copy()
remove_unidentified(images, rank="order")
pd.testing.assert_frame_equal(images_original, images)
def test_keep_index(images):
result = remove_unidentified(images, rank="order", reset_index=False)
expected_index = pd.Index([0, 1, 2, 5], dtype="int64")
pd.testing.assert_index_equal(result.index, expected_index)