def test(self): self.assertIsNotNone(COUNT.units_func) new_units = COUNT.units_func(None) self.assertEqual(new_units, 1)