def test_to_dataframe_filtering_family(self):
     """
     Test that family sensor filter works
     """
     p = network.SensorList()
     p.to_dataframe('family', 'parent')
     p.to_dataframe('family', 'child')
 def test_to_dataframe_filtering_no_value(self):
     """
     Test that providing a bad value fails
     """
     p = network.SensorList()
     p.to_dataframe('column', 'parent', 'temp_f')
     p.to_dataframe('column', 'child', 'temp_f')
 def test_to_dataframe_filtering_useful(self):
     """
     Test that useful sensor filter works
     """
     p = network.SensorList()
     p.to_dataframe('useful', 'parent')
     p.to_dataframe('useful', 'child')
 def test_to_dataframe_filtering_no_child(self):
     """
     Test that no_child sensor filter works
     """
     p = network.SensorList()
     p.to_dataframe('no_child', 'parent')
     p.to_dataframe('no_child', 'child')
 def test_to_dataframe_filtering_outside(self):
     """
     Test that outside sensor filter works
     """
     p = network.SensorList()
     p.to_dataframe('outside', 'parent')
     p.to_dataframe('outside', 'child')
 def test_to_dataframe_no_filtering(self):
     """
     Test that not using sensor filters works
     """
     p = network.SensorList()
     p.to_dataframe('all', 'parent')
     p.to_dataframe('all', 'child')
 def test_to_dataframe_filtering_good_value(self):
     """
     Test that providing a bad value fails
     """
     p = network.SensorList()
     p.to_dataframe('column', 'parent', 'location_type', 'outside')
     with self.assertRaises(ValueError):
         p.to_dataframe('column', 'child', 'location_type', 'outside')
    def test_to_dataframe_filtering_bad_column(self):
        """
        Test that providing a bad column fails
        """
        p = network.SensorList()
        with self.assertRaises(ValueError):
            p.to_dataframe('column', 'parent', 'fake_col_name')

        with self.assertRaises(ValueError):
            p.to_dataframe('column', 'child', 'fake_col_name')
 def test_setup_purpleair(self):
     """
     Test that we can initialize a SensorList
     """
     p = network.SensorList()
     self.assertIsInstance(p, network.SensorList)
 def test_to_dataframe_cols(self):
     p = network.SensorList()
     df_a = p.to_dataframe(sensor_filter='all', channel='parent')
     df_b = p.to_dataframe(sensor_filter='all', channel='child')
     self.assertListEqual(list(df_a.columns), list(df_b.columns))
 def test_outside_sensor_filtering(self):
     p = network.SensorList()
     self.assertLess(len(p.outside_sensors), len(p.all_sensors))
 def test_setup_purpleair(self):
     p = network.SensorList()
     self.assertIsInstance(p, network.SensorList)
 def test_useful_sensor_filtering(self):
     p = network.SensorList()
     self.assertLess(len(p.useful_sensors), len(p.all_sensors))