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)