-__all__ = ["unittest", "WebMockTestCase", "mock", "D",
- "StringIO"]
+__all__ = ["limits", "unittest", "WebMockTestCase", "mock", "D",
+ "StringIO", "tz"]
+
+limits = ["acceptance", "unit"]
+for test_type in limits:
+ if "--no{}".format(test_type) in sys.argv:
+ sys.argv.remove("--no{}".format(test_type))
+ limits.remove(test_type)
+ if "--only{}".format(test_type) in sys.argv:
+ sys.argv.remove("--only{}".format(test_type))
+ limits = [test_type]
+ break
+
+def tz(hours):
+ return datetime.timezone(datetime.timedelta(hours=hours))