mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-30 03:22:14 +00:00
18 lines
579 B
Python
18 lines
579 B
Python
import difflib
|
|
|
|
class TSV:
|
|
"""Helper to get pretty diffs between expected and actual tab-separated value files"""
|
|
|
|
def __init__(self, contents):
|
|
self.lines = contents.readlines() if isinstance(contents, file) else contents.splitlines(True)
|
|
|
|
def __eq__(self, other):
|
|
return self.lines == other.lines
|
|
|
|
def diff(self, other):
|
|
return list(line.rstrip() for line in difflib.context_diff(self.lines, other.lines))[2:]
|
|
|
|
@staticmethod
|
|
def toMat(contents):
|
|
return [line.split("\t") for line in contents.split("\n") if line.strip()]
|