def test_scan_leave_file_in_right_place2():
f = open(tiny_file)
result = scan(f, 8)
assert np.array_equal(result, tiny_arr[:8])
next = f.readline()
print next
print "rest of file:\n", f.read()
assert next.strip() == "some random text"
def test_big():
"""
test scanning a large enough data file to need re-allocating arrays
"""
N = 10000
f = open('junk_large.txt', 'w')
for i in range(N):
f.write("%f,\n"%3.1459)
f.close()
f = open('junk_large.txt', 'r')
result = scan(f, num_to_read=None)
assert np.array_equal( result, np.zeros((N,), dtype=np.float64)+3.1459 )
def test_assert_not_int():
f = open(tiny_file)
with pytest.raises(TypeError):
result = scan(f, 'a_string')
def test_read_negative_vals():
# it might as well work
f = open(tiny_file)
with pytest.raises(OverflowError):
result = scan(f, -3)
def test_read_zero_vals():
# it might as well work
f = open(tiny_file)
result = scan(f, 0)
assert np.array_equal(result, np.zeros((0,)))
def test_wrong_file_mode():
f = open('junk.txt', 'w')
with pytest.raises(TypeError):
result = scan(f, 10)
def test_file_closed():
f = open(tiny_file)
f.close()
with pytest.raises(TypeError):
result = scan(f, 10)
def test_assert_not_file():
with pytest.raises(TypeError):
f = scan('a_string', 4)
def test_scan_all():
f = open(tiny_file)
result = scan(f)
print result.shape
assert np.array_equal(result, tiny_arr)
def test_scan_leave_file_in_right_place():
f = open(tiny_file)
result = scan(f, 10)
next = f.readline()
assert next.strip() == "; 3"
def test_scan_not_enough():
f = open(tiny_file)
with pytest.raises(ValueError):
result = scan(f, 15)
def test_scan_n():
f = open(tiny_file)
result = scan(f, 10)
assert np.array_equal(tiny_arr[:10], result)
def test_call():
" can we even call it"
f = open(tiny_file)
result = scan(f, 10)
print result
assert True
