|
1 | 1 | class TLCGreenTaxiTripTest < Test::Unit::TestCase
|
2 |
| - test("each") do |
3 |
| - dataset = DatasetsParquet::TLC::GreenTaxiTrip.new(year: 2022, month: 1) |
4 |
| - records = dataset.each.to_a |
| 2 | + def setup |
| 3 | + @default_timezone_env = ENV['TZ'] |
| 4 | + ENV['TZ'] = 'UTC' |
| 5 | + @dataset = DatasetsParquet::TLC::GreenTaxiTrip.new(year: 2022, month: 1) |
| 6 | + end |
| 7 | + |
| 8 | + def teardown |
| 9 | + ENV['TZ'] = @default_timezone_env |
| 10 | + end |
| 11 | + |
| 12 | + test("#to_arrow") do |
| 13 | + assert_equal(<<~TABLE, @dataset.to_arrow.to_s) |
| 14 | +\tVendorID\t lpep_pickup_datetime\t lpep_dropoff_datetime\tstore_and_fwd_flag\tRatecodeID\tPULocationID\tDOLocationID\tpassenger_count\ttrip_distance\tfare_amount\t extra\t mta_tax\ttip_amount\ttolls_amount\tehail_fee\timprovement_surcharge\ttotal_amount\tpayment_type\t trip_type\tcongestion_surcharge |
| 15 | + 0\t 2\t2022-01-01T00:14:21+00:00\t2022-01-01T00:15:33+00:00\tN \t 1.000000\t 42\t 42\t 1.000000\t 0.440000\t 3.500000\t 0.500000\t 0.500000\t 0.000000\t 0.000000\t (null)\t 0.300000\t 4.800000\t 2.000000\t 1.000000\t 0.000000 |
| 16 | + 1\t 1\t2022-01-01T00:20:55+00:00\t2022-01-01T00:29:38+00:00\tN \t 1.000000\t 116\t 41\t 1.000000\t 2.100000\t 9.500000\t 0.500000\t 0.500000\t 0.000000\t 0.000000\t (null)\t 0.300000\t 10.800000\t 2.000000\t 1.000000\t 0.000000 |
| 17 | + 2\t 1\t2022-01-01T00:57:02+00:00\t2022-01-01T01:13:14+00:00\tN \t 1.000000\t 41\t 140\t 1.000000\t 3.700000\t 14.500000\t 3.250000\t 0.500000\t 4.600000\t 0.000000\t (null)\t 0.300000\t 23.150000\t 1.000000\t 1.000000\t 2.750000 |
| 18 | + 3\t 2\t2022-01-01T00:07:42+00:00\t2022-01-01T00:15:57+00:00\tN \t 1.000000\t 181\t 181\t 1.000000\t 1.690000\t 8.000000\t 0.500000\t 0.500000\t 0.000000\t 0.000000\t (null)\t 0.300000\t 9.300000\t 2.000000\t 1.000000\t 0.000000 |
| 19 | + 4\t 2\t2022-01-01T00:07:50+00:00\t2022-01-01T00:28:52+00:00\tN \t 1.000000\t 33\t 170\t 1.000000\t 6.260000\t 22.000000\t 0.500000\t 0.500000\t 5.210000\t 0.000000\t (null)\t 0.300000\t 31.260000\t 1.000000\t 1.000000\t 2.750000 |
| 20 | + 5\t 1\t2022-01-01T00:47:57+00:00\t2022-01-01T00:54:09+00:00\tN \t 1.000000\t 150\t 210\t 1.000000\t 1.300000\t 7.000000\t 0.500000\t 0.500000\t 0.000000\t 0.000000\t (null)\t 0.300000\t 8.300000\t 2.000000\t 1.000000\t 0.000000 |
| 21 | + 6\t 2\t2022-01-01T00:13:38+00:00\t2022-01-01T00:33:50+00:00\tN \t 1.000000\t 66\t 67\t 1.000000\t 6.470000\t 22.500000\t 0.500000\t 0.500000\t 0.000000\t 0.000000\t (null)\t 0.300000\t 23.800000\t 2.000000\t 1.000000\t 0.000000 |
| 22 | + 7\t 2\t2022-01-01T00:43:00+00:00\t2022-01-01T00:49:20+00:00\tN \t 1.000000\t 40\t 195\t 1.000000\t 1.150000\t 6.000000\t 0.500000\t 0.500000\t 0.000000\t 0.000000\t (null)\t 0.300000\t 7.300000\t 2.000000\t 1.000000\t 0.000000 |
| 23 | + 8\t 2\t2022-01-01T00:41:04+00:00\t2022-01-01T00:47:04+00:00\tN \t 1.000000\t 112\t 80\t 1.000000\t 1.300000\t 6.000000\t 0.500000\t 0.500000\t 0.000000\t 0.000000\t (null)\t 0.300000\t 7.300000\t 2.000000\t 1.000000\t 0.000000 |
| 24 | + 9\t 2\t2022-01-01T00:51:07+00:00\t2022-01-01T01:09:31+00:00\tN \t 1.000000\t 256\t 186\t 1.000000\t 4.750000\t 17.000000\t 0.500000\t 0.500000\t 4.210000\t 0.000000\t (null)\t 0.300000\t 25.260000\t 1.000000\t 1.000000\t 2.750000 |
| 25 | +... |
| 26 | +62485\t 2\t2022-01-31T22:01:00+00:00\t2022-01-31T22:13:00+00:00\t (null)\t (null)\t 244\t 151\t (null)\t 4.690000\t 19.010000\t 0.000000\t 0.000000\t 4.120000\t 0.000000\t (null)\t 0.300000\t 26.180000\t (null)\t (null)\t (null) |
| 27 | +62486\t 2\t2022-01-31T22:54:00+00:00\t2022-01-31T23:10:00+00:00\t (null)\t (null)\t 25\t 188\t (null)\t 3.120000\t 13.900000\t 0.000000\t 0.000000\t 3.160000\t 0.000000\t (null)\t 0.300000\t 17.360000\t (null)\t (null)\t (null) |
| 28 | +62487\t 2\t2022-01-31T23:23:00+00:00\t2022-01-31T23:39:00+00:00\t (null)\t (null)\t 179\t 112\t (null)\t 3.800000\t 16.480000\t 0.000000\t 0.000000\t 3.730000\t 0.000000\t (null)\t 0.300000\t 20.510000\t (null)\t (null)\t (null) |
| 29 | +62488\t 2\t2022-01-31T23:50:00+00:00\t2022-02-01T00:11:00+00:00\t (null)\t (null)\t 112\t 239\t (null)\t 6.040000\t 23.450000\t 0.000000\t 0.000000\t 5.830000\t 0.000000\t (null)\t 0.300000\t 32.330000\t (null)\t (null)\t (null) |
| 30 | +62489\t 2\t2022-01-31T23:19:00+00:00\t2022-01-31T23:37:00+00:00\t (null)\t (null)\t 152\t 233\t (null)\t 6.710000\t 25.400000\t 0.000000\t 0.000000\t 6.270000\t 0.000000\t (null)\t 0.300000\t 34.720000\t (null)\t (null)\t (null) |
| 31 | +62490\t 2\t2022-01-31T23:25:00+00:00\t2022-01-31T23:33:00+00:00\t (null)\t (null)\t 40\t 65\t (null)\t 1.400000\t 8.380000\t 0.000000\t 0.000000\t 1.930000\t 0.000000\t (null)\t 0.300000\t 10.610000\t (null)\t (null)\t (null) |
| 32 | +62491\t 2\t2022-01-31T23:52:00+00:00\t2022-02-01T00:10:00+00:00\t (null)\t (null)\t 36\t 61\t (null)\t 2.970000\t 14.920000\t 0.000000\t 0.000000\t 0.000000\t 0.000000\t (null)\t 0.300000\t 15.220000\t (null)\t (null)\t (null) |
| 33 | +62492\t 2\t2022-01-31T23:17:00+00:00\t2022-01-31T23:36:00+00:00\t (null)\t (null)\t 75\t 167\t (null)\t 3.700000\t 16.260000\t 0.000000\t 0.000000\t 0.000000\t 0.000000\t (null)\t 0.300000\t 16.560000\t (null)\t (null)\t (null) |
| 34 | +62493\t 2\t2022-01-31T23:45:00+00:00\t2022-01-31T23:55:00+00:00\t (null)\t (null)\t 116\t 166\t (null)\t 1.880000\t 9.480000\t 0.000000\t 0.000000\t 2.170000\t 0.000000\t (null)\t 0.300000\t 11.950000\t (null)\t (null)\t (null) |
| 35 | +62494\t 2\t2022-01-31T23:52:00+00:00\t2022-02-01T00:26:00+00:00\t (null)\t (null)\t 225\t 179\t (null)\t 9.600000\t 32.180000\t 0.000000\t 0.000000\t 7.230000\t 10.000000\t (null)\t 0.300000\t 49.710000\t (null)\t (null)\t (null) |
| 36 | + TABLE |
| 37 | + end |
| 38 | + |
| 39 | + test("#each") do |
| 40 | + records = @dataset.each.to_a |
5 | 41 |
|
6 | 42 | assert_equal([
|
7 | 43 | 62495,
|
|
0 commit comments