1. import re
  3. def assert_text_blocks(text, assertions):
  4. """
  5. Asserts conditions of text blocks.
  7. Args:
  8. text: The text to analyze.
  9. assertions: A list of dictionaries, where each dictionary represents an assertion.
  10. Each dictionary should have the following keys:
  11. 'pattern': A regular expression pattern to match.
  12. 'condition': A function that takes a match object and returns True if the condition is met, False otherwise.
  13. 'message': A message to display if the assertion fails.
  14. """
  15. for assertion in assertions:
  16. pattern = assertion['pattern']
  17. condition = assertion['condition']
  18. message = assertion['message']
  20. match = re.search(pattern, text)
  21. if match:
  22. if not condition(match):
  23. raise AssertionError(message)
  24. else:
  25. raise AssertionError(message)
  27. if __name__ == '__main__':
  28. # Example Usage
  29. text = "The price is $100.00 and the quantity is 2."
  30. assertions = [
  31. {'pattern': r'\$\d+\.\d{2}', 'condition': lambda match: float(match.group(0)) > 50, 'message': 'Price must be greater than 50'},
  32. {'pattern': r'\d+', 'condition': lambda match: int(match.group(0)) > 0, 'message': 'Quantity must be positive'}
  33. ]
  35. try:
  36. assert_text_blocks(text, assertions)
  37. print("All assertions passed!")
  38. except AssertionError as e:
  39. print(f"Assertion failed: {e}")
  41. text2 = "The price is $25.00 and the quantity is 0."
  42. assertions2 = [
  43. {'pattern': r'\$\d+\.\d{2}', 'condition': lambda match: float(match.group(0)) > 50, 'message': 'Price must be greater than 50'},
  44. {'pattern': r'\d+', 'condition': lambda match: int(match.group(0)) > 0, 'message': 'Quantity must be positive'}
  45. ]
  47. try:
  48. assert_text_blocks(text2, assertions2)
  49. print("All assertions passed!")
  50. except AssertionError as e:
  51. print(f"Assertion failed: {e}")

Add your comment