Skip to content
Draft
Show file tree
Hide file tree
Changes from 6 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 10 additions & 7 deletions crates/ruff_memory_usage/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -16,13 +16,16 @@ pub fn heap_size<T: GetSize>(value: &T) -> usize {

/// An implementation of [`GetSize::get_heap_size`] for [`OrderSet`].
pub fn order_set_heap_size<T: GetSize, S>(set: &OrderSet<T, S>) -> usize {
(set.capacity() * T::get_stack_size()) + set.iter().map(heap_size).sum::<usize>()
let size = set.iter().map(heap_size::<T>).sum::<usize>();
size + (set.capacity() * T::get_stack_size())
}

/// An implementation of [`GetSize::get_heap_size`] for [`OrderMap`].
pub fn order_map_heap_size<K: GetSize, V: GetSize, S>(map: &OrderMap<K, V, S>) -> usize {
(map.capacity() * (K::get_stack_size() + V::get_stack_size()))
+ (map.iter())
.map(|(k, v)| heap_size(k) + heap_size(v))
.sum::<usize>()
/// An implementation of [`GetSize::get_heap_size`] for [`OrderSet`].
pub fn order_map_heap_size<K: GetSize, V: GetSize, S>(set: &OrderMap<K, V, S>) -> usize {
let size = set
.iter()
.map(|(key, val)| heap_size::<K>(key) + heap_size::<V>(val))
.sum::<usize>();

size + (set.capacity() * <(K, V)>::get_stack_size())
}
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
# Unsupported special types

We do not understand the functional syntax for creating `NamedTuple`s, `TypedDict`s or `Enum`s yet.
But we also do not emit false positives when these are used in type expressions.
We do not understand the functional syntax for creating `NamedTuple`s or `Enum`s yet. But we also do
not emit false positives when these are used in type expressions.

```py
import collections
Expand All @@ -10,9 +10,8 @@ import typing

MyEnum = enum.Enum("MyEnum", ["foo", "bar", "baz"])
MyIntEnum = enum.IntEnum("MyIntEnum", ["foo", "bar", "baz"])
MyTypedDict = typing.TypedDict("MyTypedDict", {"foo": int})
MyNamedTuple1 = typing.NamedTuple("MyNamedTuple1", [("foo", int)])
MyNamedTuple2 = collections.namedtuple("MyNamedTuple2", ["foo"])

def f(a: MyEnum, b: MyTypedDict, c: MyNamedTuple1, d: MyNamedTuple2): ...
def f(a: MyEnum, c: MyNamedTuple1, d: MyNamedTuple2): ...
```
Loading