IF YOU WOULD LIKE TO GET AN ACCOUNT, please write an email to s dot adaszewski at gmail dot com. User accounts are meant only to report issues and/or generate pull requests. This is a purpose-specific Git hosting for ADARED projects. Thank you for your understanding!
Browse Source

Add test_cumcount.

master
Stanislaw Adaszewski 3 years ago
parent
commit
cee0c23bd7
1 changed files with 26 additions and 0 deletions
  1. +26
    -0
      tests/triacontagon/test_cumcount.py

+ 26
- 0
tests/triacontagon/test_cumcount.py View File

@@ -0,0 +1,26 @@
from triacontagon.cumcount import dfill, \
argunsort, \
cumcount
import numpy as np
def test_dfill_01():
input = np.array([1, 1, 1, 1, 1, 2, 2, 3, 3, 3, 4, 4, 5, 5])
output = dfill(input)
expected = np.array([0, 0, 0, 0, 0, 5, 5, 7, 7, 7, 10, 10, 12, 12])
assert np.all(output == expected)
def test_argunsort_01():
input = np.array([1, 1, 2, 3, 3, 4, 1, 1, 5, 5, 2, 3, 1, 4])
output = np.argsort(input, kind='mergesort')
output = argunsort(output)
expected = np.array([0, 1, 5, 7, 8, 10, 2, 3, 12, 13, 6, 9, 4, 11])
assert np.all(output == expected)
def test_cumcount_01():
input = np.array([1, 1, 2, 3, 3, 4, 1, 1, 5, 5, 2, 3, 1, 4])
output = cumcount(input)
expected = np.array([0, 1, 0, 0, 1, 0, 2, 3, 0, 1, 1, 2, 4, 1])
assert np.all(output == expected)

Loading…
Cancel
Save