def test_round_trip_int_nr(fname): # test data np.random.seed(0) edges = np.arange(100) vals = np.arange(100) # write to disk sn = csv_dist_sink(fname) sn.activate() sn.write_dist(edges, vals) sn.deactivate() # read from disk sr = csv_dist_source(fname) sr.activate() bin_edges = sr.bin_edges() read_vals = sr.values() sr.deactivate() assert_almost_equal(read_vals, vals) assert_almost_equal(bin_edges, edges)
def test_src_pickle(fname): sr = csv_dist_source(fname) ck = pickle.dumps(sr) pickle.loads(ck)
def test_src_active(fname): # write to disk sn = csv_dist_source(fname) sn.values()