def test_csv_row_iterator():
print(os.getcwd()) # If test is successful this does not get printed to stdout
# py.test looks for a file to open in the project folder rather than the
# package folder unlike the case when we run the module itself, hence the
# path to the file to open needs to be specified in this manner.
with open(os.getcwd() + '/features/anscombe.txt') as source:
rows = tuple(csv_row_iterator(source, '\t'))
assert(len(rows) == 14)
def test_series_point_generator():
with open(os.getcwd() + '/features/anscombe.txt') as source:
data = tuple(csv_row_trim_iterator(csv_row_iterator(source)))
series_0 = tuple(series_point_generator(0, data))
assert(len(series_0) == 11)
assert(len(series_0[0]) == 2)
series_1 = tuple(series_point_generator(1, data))
assert(len(series_1) == 11)
assert(len(series_1[0]) == 2)
series_2 = tuple(series_point_generator(2, data))
assert(len(series_2) == 11)
assert(len(series_2[0]) == 2)
series_3 = tuple(series_point_generator(3, data))
assert(len(series_3) == 11)
assert(len(series_3[0]) == 2)
def test_csv_head_trimmer():
with open(os.getcwd() + '/features/anscombe.txt') as source:
rows = tuple(csv_row_trim_iterator(csv_row_iterator(source, '\t')))
assert(len(rows) == 11)
