def test_remove_antenna_list_3(self):
data = uf.remove_antennas(self.data, ["1", "2"])
mice = set(data["Tag"])
self.assertEqual(mice, set(["mouse_1", "mouse_2"]))
def test_remove_antenna_list_1(self):
data = uf.remove_antennas(self.data, ["1", None])
self.assertEqual(data.shape[0], self.data.shape[0] - 3)
def test_remove_antenna_list_2_1(self):
data = uf.remove_antennas(self.data, ["1", "2"])
self.assertEqual(self.data.shape[0] - 6, data.shape[0])
def test_single_antenna_1(self):
data = uf.remove_antennas(self.data, "1")